tf_agents.policies.utils.get_model_index

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-04-26 UTC.

tf_agents.policies.utils.get_model_index

Args

Returns