Would it be possible to use `metrics.mean_squared_error` or `metrics.mean_absolute_error`?