reagent.net_builder.discrete_dqn package

Submodules

reagent.net_builder.discrete_dqn.dueling module

class reagent.net_builder.discrete_dqn.dueling.Dueling(sizes: List[int] = <factory>, activations: List[str] = <factory>)

Bases: reagent.net_builder.discrete_dqn_net_builder.DiscreteDQNNetBuilder

activations: List[str]
build_q_network(state_feature_config: reagent.core.types.ModelFeatureConfig, state_normalization_data: reagent.core.parameters.NormalizationData, output_dim: int) reagent.models.base.ModelBase
sizes: List[int]

reagent.net_builder.discrete_dqn.fully_connected module

class reagent.net_builder.discrete_dqn.fully_connected.FullyConnected(sizes: List[int] = <factory>, activations: List[str] = <factory>, dropout_ratio: float = 0.0, use_batch_norm: bool = False)

Bases: reagent.net_builder.discrete_dqn_net_builder.DiscreteDQNNetBuilder

activations: List[str]
build_q_network(state_feature_config: reagent.core.types.ModelFeatureConfig, state_normalization_data: reagent.core.parameters.NormalizationData, output_dim: int) reagent.models.base.ModelBase
dropout_ratio: float = 0.0
sizes: List[int]
use_batch_norm: bool = False

reagent.net_builder.discrete_dqn.fully_connected_with_embedding module

class reagent.net_builder.discrete_dqn.fully_connected_with_embedding.FullyConnectedWithEmbedding(sizes: List[int] = <factory>, activations: List[str] = <factory>, embedding_dim: int = 64, dropout_ratio: float = 0.0)

Bases: reagent.net_builder.discrete_dqn_net_builder.DiscreteDQNNetBuilder

activations: List[str]
build_q_network(state_feature_config: reagent.core.types.ModelFeatureConfig, state_normalization_data: reagent.core.parameters.NormalizationData, output_dim: int) reagent.models.base.ModelBase
dropout_ratio: float = 0.0
embedding_dim: int = 64
sizes: List[int]

Module contents