d3rlpy.metrics.scorer.dynamics_prediction_variance_scorer

d3rlpy.metrics.scorer.dynamics_prediction_variance_scorer(dynamics, episodes)[source]

Returns prediction variance of ensemble dynamics.

This metrics suggests how dynamics model is confident of test sets. If the variance is large, the dynamics model has large uncertainty.

Parameters
  • dynamics (d3rlpy.metrics.scorer.DynamicsProtocol) – dynamics model.

  • episodes (List[d3rlpy.dataset.Episode]) – list of episodes.

Returns

variance.

Return type

float