sourav11295 commited on
Commit
a74ed04
·
1 Parent(s): 6471834

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -113,12 +113,12 @@ def regression(df,dep):
113
  },
114
  ]
115
 
116
- search = GridSearchCV(pipe, param_grid=parameters, cv=5, n_jobs=-1, scoring='neg_mean_absolute_percentage_error')
117
  search.fit(X_train,y_train)
118
 
119
  result = pd.DataFrame(search.cv_results_)[['params','rank_test_score','mean_test_score']]
120
 
121
- result['mean_test_score']= (result['mean_test_score']+1)*100
122
  result = result.astype({'params': str})
123
 
124
  result.sort_values('rank_test_score',inplace=True)
 
113
  },
114
  ]
115
 
116
+ search = GridSearchCV(pipe, param_grid=parameters, cv=5, n_jobs=-1, scoring='r2')
117
  search.fit(X_train,y_train)
118
 
119
  result = pd.DataFrame(search.cv_results_)[['params','rank_test_score','mean_test_score']]
120
 
121
+ result['mean_test_score']= (result['mean_test_score'])*100
122
  result = result.astype({'params': str})
123
 
124
  result.sort_values('rank_test_score',inplace=True)