# tf_agents.policies.utils.get_model_index

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/policies/utils.py#L339-L354) |

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.

<br />

<br />

<br />

| ## 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. |

<br />

<br />

<br />

<br />

| ## Returns ||
|---|---|
| The index of the model for the arm requested. ||

<br />