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 |
|