Commit 2344bd2d authored by Susheel Busi's avatar Susheel Busi
Browse files

added print in grid_search

parent 7e7aecb9
......@@ -74,6 +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():
print(param)
res[param] = best_params[param.split('__')[1]]
for metric in metrics:
res[metric] = scores[metric]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment