scripts/grid_search.py

@@ -68,16 +68,16 @@ def grid_search(X, y, model, emb_func, trans_func, param_grid, scorings, cv, cor
     scores = cross_validate(grid_search, X, y, scoring=scorings, n_jobs=cores, cv=outer_cv)
 
     # store the result in a dataframe
-    best_params = grid_search.best_estimator_
+    best_params = grid_search.best_params_
     print(best_params)
     print(scores)
     metrics = list(scorings.keys())
     res = dict()
-    for param in param_grid.keys():
-        print(param)
-        res[param] = best_params[param.split('__')[1]]
+    for param in best_params.keys():
+        res[param] = best_params[param]
     for metric in metrics:
         res[metric] = scores[metric]
     df = pd.DataFrame.from_dict(res)
     return df
 
+