Returns the model index for a specific arm.
tf_agents.policies.utils.get_model_index(
arm_index: int, accepts_per_arm_features: bool
) -> int
The number of models depends on the observation format: If the policy accepts
per-arm features, there is only one single model used for every arm. Otherwise
there is a model for every arm.
Args |
|---|
arm_index
|
The index of the arm for which the model index is needed.
|
accepts_per_arm_features
|
(bool) Whether the policy works with per-arm
features.
|
Returns |
|---|
|
The index of the model for the arm requested.
|
View source on GitHub