Spaces:
Runtime error
Runtime error
Commit
·
a74ed04
1
Parent(s):
6471834
Update app.py
Browse files
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='
|
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']
|
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)
|