Skip to content

Commit 66faa35

Browse files
Update mean_squared_error.py
1 parent 6f26bc3 commit 66faa35

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

machine_learning/losses/mean_squared_error.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,3 +25,8 @@ def mean_squared_error(y_true: np.ndarray, y_pred: np.ndarray) -> float:
2525
mse = np.mean(squared_errors)
2626

2727
return mse
28+
29+
if __name__ == "__main__":
30+
import doctest
31+
32+
doctest.testmod()

0 commit comments

Comments
 (0)