commit-live-students · ms10031990 · Feb 22, 2018
diff --git a/__init__.pyc b/__init__.pyc
diff --git a/q01_grid_search/__init__.pyc b/q01_grid_search/__init__.pyc
diff --git 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
diff --git a/q01_grid_search/tests/__init__.pyc b/q01_grid_search/tests/__init__.pyc
diff --git a/q01_grid_search/tests/test_q01_grid_search.pyc b/q01_grid_search/tests/test_q01_grid_search.pyc