tensorly.metrics.regression.RMSE

RMSE(y_true, y_pred, axis=None)[source]

Returns the regularised mean squared error between the two predictions (the square-root is applied to the mean_squared_error)

Parameters:

y_truearray of shape (n_samples, )

Ground truth (correct) target values.

y_predarray of shape (n_samples, )

Estimated target values.

Returns:

float