reagent.net_builder.categorical_dqn package
Submodules
reagent.net_builder.categorical_dqn.categorical module
- class reagent.net_builder.categorical_dqn.categorical.Categorical(sizes: List[int] = <factory>, activations: List[str] = <factory>)
Bases:
reagent.net_builder.categorical_dqn_net_builder.CategoricalDQNNetBuilder- activations: List[str]
- build_q_network(state_normalization_data: reagent.core.parameters.NormalizationData, output_dim: int, num_atoms: int, qmin: int, qmax: int) reagent.models.base.ModelBase
- sizes: List[int]