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]