GridSearchCV warm-up
In the video we saw that increasing the RBF kernel hyperparameter gamma
increases training accuracy. In this exercise we'll search for the gamma
that maximizes cross-validation accuracy using scikit-learn's GridSearchCV
. A binary version of the handwritten digits dataset, in which you're just trying to predict whether or not an image is a "2", is already loaded into the variables X
and y
.
This exercise is part of the course
Linear Classifiers in Python
Exercise instructions
- Create a
GridSearchCV
object. - Call the
fit()
method to select the best value ofgamma
based on cross-validation accuracy.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Instantiate an RBF SVM
svm = SVC()
# Instantiate the GridSearchCV object and run the search
parameters = {'gamma':[0.00001, 0.0001, 0.001, 0.01, 0.1]}
searcher = GridSearchCV(svm, ____)
____.fit(____)
# Report the best parameters
print("Best CV params", searcher.best_params_)