Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
| 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 |
|