diff --git a/__init__.pyc b/__init__.pyc index 9f1c574..7a764ef 100644 Binary files a/__init__.pyc and b/__init__.pyc differ diff --git a/q01_grid_search/__init__.pyc b/q01_grid_search/__init__.pyc index 545de0a..7a2f036 100644 Binary files a/q01_grid_search/__init__.pyc and b/q01_grid_search/__init__.pyc differ diff --git a/q01_grid_search/build.py b/q01_grid_search/build.py index 1438657..10fabeb 100644 --- a/q01_grid_search/build.py +++ b/q01_grid_search/build.py @@ -18,7 +18,12 @@ "max_leaf_nodes": [5, 10, 2]} -# Write your solution here : - - - +def grid_search(X_train, y_train, model, param_grid, cv=3): + clf = GridSearchCV(model,param_grid,cv=cv) + clf.fit(X_train,y_train) + + results = pd.DataFrame( clf.grid_scores_) + model = clf + variable1 = tuple(results['parameters']) + variable2 = results['mean_validation_score'].as_matrix() # Returns numpy array + return model, variable1 , variable2 diff --git a/q01_grid_search/build.pyc b/q01_grid_search/build.pyc index 4470232..cbd029b 100644 Binary files a/q01_grid_search/build.pyc and b/q01_grid_search/build.pyc differ diff --git a/q01_grid_search/tests/__init__.pyc b/q01_grid_search/tests/__init__.pyc index 426e901..b823c38 100644 Binary files a/q01_grid_search/tests/__init__.pyc and b/q01_grid_search/tests/__init__.pyc differ diff --git a/q01_grid_search/tests/test_q01_grid_search.pyc b/q01_grid_search/tests/test_q01_grid_search.pyc index 782fd4f..f77afcd 100644 Binary files a/q01_grid_search/tests/test_q01_grid_search.pyc and b/q01_grid_search/tests/test_q01_grid_search.pyc differ