Save the date! Google I/O returns May 18-20 Register now


Returns the model index for a specific arm.

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.

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.

The index of the model for the arm requested.