Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ from statsforecast.models import (
|
|
| 22 |
)
|
| 23 |
|
| 24 |
from utilsforecast.evaluation import evaluate
|
| 25 |
-
from
|
| 26 |
|
| 27 |
# Import for MLForecast
|
| 28 |
from mlforecast import MLForecast
|
|
@@ -321,11 +321,14 @@ def run_forecast(
|
|
| 321 |
y_true_clean = y_true[mask]
|
| 322 |
y_pred_clean = y_pred[mask]
|
| 323 |
|
|
|
|
|
|
|
|
|
|
| 324 |
metrics_list.append({
|
| 325 |
'Model': col,
|
| 326 |
-
'MAE':
|
| 327 |
-
'RMSE':
|
| 328 |
-
'MAPE':
|
| 329 |
})
|
| 330 |
|
| 331 |
eval_metrics = pd.DataFrame(metrics_list)
|
|
|
|
| 22 |
)
|
| 23 |
|
| 24 |
from utilsforecast.evaluation import evaluate
|
| 25 |
+
from sklearn.metrics import mean_absolute_error, mean_squared_error, mean_absolute_percentage_error
|
| 26 |
|
| 27 |
# Import for MLForecast
|
| 28 |
from mlforecast import MLForecast
|
|
|
|
| 321 |
y_true_clean = y_true[mask]
|
| 322 |
y_pred_clean = y_pred[mask]
|
| 323 |
|
| 324 |
+
# Calculate RMSE manually
|
| 325 |
+
rmse_value = np.sqrt(mean_squared_error(y_true_clean, y_pred_clean))
|
| 326 |
+
|
| 327 |
metrics_list.append({
|
| 328 |
'Model': col,
|
| 329 |
+
'MAE': mean_absolute_error(y_true_clean, y_pred_clean),
|
| 330 |
+
'RMSE': rmse_value,
|
| 331 |
+
'MAPE': mean_absolute_percentage_error(y_true_clean, y_pred_clean) * 100
|
| 332 |
})
|
| 333 |
|
| 334 |
eval_metrics = pd.DataFrame(metrics_list)
|