scripts/grid_search.py

@@ -74,7 +74,7 @@ def grid_search(X, y, model, emb_func, trans_func, param_grid, scorings, cv, cor
     metrics = list(scorings.keys())
     res = dict()
     for param in param_grid.keys():
-        res[param] = best_params[param]
+        res[param] = best_params[param.split('__')[1]]
     for metric in metrics:
         res[metric] = scores[metric]
     df = pd.DataFrame.from_dict(res)