avalanche.evaluation.metrics.TaskAwareRMSE

class avalanche.evaluation.metrics.TaskAwareRMSE[source]

The task-aware RMSE metric.

The metric computes a dictionary of <task_label, RMSE value> pairs. update/result/reset methods are all task-aware.

__init__()[source]

Creates an instance of the task-aware RMSE metric.

Methods

__init__()

Creates an instance of the task-aware RMSE metric.

reset([task_label])

Resets the metric.

result([task_label])

Retrieves the running RMSE.

update(predicted_y, true_y, task_labels)

Update the running RMSE given the true and predicted labels.