JoBeer commited on
Commit
5a4f664
·
verified ·
1 Parent(s): 7e81330

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -104,7 +104,7 @@ def test(model_fit,test,train,df_covariates,time):
104
  for i in df_prediction.columns:
105
  target_variable.append(i)
106
  MAE.append(mean_absolute_error(df_prediction[i], test.to_dataframe()[i]))
107
- R2.append(r2_score(df_prediction[i], test.to_dataframe()[i]))
108
  metrics = pd.DataFrame({'target':target_variable,'MAE': MAE, 'R2': R2})
109
  return metrics, fig
110
 
 
104
  for i in df_prediction.columns:
105
  target_variable.append(i)
106
  MAE.append(mean_absolute_error(df_prediction[i], test.to_dataframe()[i]))
107
+ R2.append(r2_score(test.to_dataframe()[i],df_prediction[i]))
108
  metrics = pd.DataFrame({'target':target_variable,'MAE': MAE, 'R2': R2})
109
  return metrics, fig
110