Skip to content

Commit 0072c4c

Browse files
committed
Adding RMSE - Root Mean Squared Error Loss function for ML Evaluation
1 parent 9ae7622 commit 0072c4c

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

machine_learning/loss_functions.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -665,28 +665,28 @@ def kullback_leibler_divergence(y_true: np.ndarray, y_pred: np.ndarray) -> float
665665
def root_mean_squared_error(y_true, y_pred):
666666
"""
667667
Root Mean Squared Error (RMSE)
668+
669+
Root Mean Squared Error (RMSE) is a standard metric used to evaluate
670+
the accuracy of regression models.
671+
It measures the average magnitude of the prediction errors, giving
672+
higher weight to larger errors due to squaring.
668673
669-
Root Mean Squared Error (RMSE) is a standard metric used to evaluate the accuracy of regression models.
670-
It measures the average magnitude of the prediction errors, giving higher weight to larger errors due to squaring.
671-
The RMSE value is always non-negative, and a lower RMSE indicates better model performance.
672-
673-
RMSE = sqrt( (1/n) * Σ (y_true - y_pred) ^ 2)
674+
RMSE = sqrt( (1/n) * Σ (y_true - y_pred) ^ 2)
674675
675676
Reference: https://en.wikipedia.org/wiki/Root_mean_square_deviation
676677
677678
Parameters:
678679
y_pred: Predicted Value
679680
y_true: Actual Value
680-
681+
681682
Returns:
682-
float: The RMSE Loss function between y_Pred and y_true
683+
float: The RMSE Loss function between y_pred and y_true
683684
684685
Example:
685686
>>> y_true = np.array([100, 200, 300])
686687
>>> y_pred = np.array([110, 190, 310])
687-
>>> rmse(A_t, F_t)
688+
>>> rmse(y_true, y_pred)
688689
3.42
689-
690690
"""
691691
y_true, y_pred = np.array(y_true), np.array(y_pred)
692692

0 commit comments

Comments
 (0)