libbeyfox commited on
Commit
1752d14
·
verified ·
1 Parent(s): 16986b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -22,7 +22,7 @@ from statsforecast.models import (
22
  )
23
 
24
  from utilsforecast.evaluation import evaluate
25
- from utilsforecast.losses import *
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': mae(y_true_clean, y_pred_clean),
327
- 'RMSE': rmse(y_true_clean, y_pred_clean),
328
- 'MAPE': mape(y_true_clean, y_pred_clean)
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)