|
8 | 8 |
|
9 | 9 | import numpy as np |
10 | 10 | from sklearn.metrics import mean_absolute_error as mae |
| 11 | +from sklearn.metrics import root_mean_squared_error as rmse |
11 | 12 | from sklearn.metrics import mean_squared_error as mse |
| 13 | +from sklearn.metrics import root_mean_squared_log_error as rmsle |
12 | 14 | from sklearn.metrics import mean_squared_log_error as msle |
13 | 15 | from sklearn.metrics import r2_score as r2 |
14 | 16 | from sklearn.metrics import roc_auc_score as auc |
@@ -86,17 +88,24 @@ def _percentage_error(real, pred): |
86 | 88 | # mae imported from sklearn |
87 | 89 | # ------------------------------------------------------------------------------ |
88 | 90 |
|
89 | | -def rmse(real, pred): |
90 | | - return mse(real, pred, squared = False) |
| 91 | +# rmse imported from sklearn |
91 | 92 | # ------------------------------------------------------------------------------ |
92 | 93 |
|
93 | | -def rmsle(real, pred): |
94 | | - '''Changes negative predictions to 0 for correct calculation''' |
95 | | - try: |
96 | | - return msle(real, pred, squared = False) |
97 | | - except ValueError: |
98 | | - pred_non_negative, real_non_negative = remove_negatives(pred, real) |
99 | | - return msle(real_non_negative, pred_non_negative, squared = False) |
| 94 | +# rmsle imported from sklearn |
| 95 | +# ------------------------------------------------------------------------------ |
| 96 | + |
| 97 | +# def rmse(real, pred): |
| 98 | +# return mse(real, pred, squared = False) |
| 99 | +# ------------------------------------------------------------------------------ |
| 100 | + |
| 101 | +# def rmsle(real, pred): |
| 102 | +# '''Changes negative predictions to 0 for correct calculation''' |
| 103 | +# try: |
| 104 | +# return msle(real, pred, squared = False) |
| 105 | +# except ValueError: |
| 106 | +# pred_non_negative, real_non_negative = remove_negatives(pred, real) |
| 107 | +# return msle(real_non_negative, pred_non_negative, squared = False) |
| 108 | + |
100 | 109 |
|
101 | 110 | # ------------------------------------------------------------------------------ |
102 | 111 |
|
|
0 commit comments