nueramic_mathml.ml.metrics.r2_score

r2_score(y_true, y_pred)

Return R2 metric of regression

(1)\[\mathbf{R}^{2} = 1 - \frac{\sum_{i=1}^{n}(\hat y_i - y_i)^2}{\sum_{i = 1}^{n}(\hat y_i - \overline y)^2}\]

Note

if std(y_true) = 0, then r2 = 0

Parameters:

• y_true (Tensor) – array with true values of regression

• y_pred (Tensor) – array with prediction values of regression

Returns:

r2 metric in float number

Return type:

float