avalanche.evaluation.metrics.rmse_metrics

avalanche.evaluation.metrics.rmse_metrics(*, minibatch=False, epoch=False, epoch_running=False, experience=False, stream=False, trained_experience=False) → List[RMSEPluginMetric]

Helper method that can be used to obtain the desired set of plugin metrics.

Parameters:

• minibatch – If True, will return a metric able to log the minibatch RMSE at training time.

• epoch – If True, will return a metric able to log the epoch RMSE at training time.

• epoch_running – If True, will return a metric able to log the running epoch RMSE at training time.

• experience – If True, will return a metric able to log the RMSE on each evaluation experience.

• stream – If True, will return a metric able to log the RMSE averaged over the entire evaluation stream of experiences.

• trained_experience – If True, will return a metric able to log the average evaluation RMSE only for experiences that the model has been trained on

Returns:

A list of plugin metrics.