python-scikit-learnHow to get best estimator from grid search CV
from sklearn import svm, datasets, model_selection
iris = datasets.load_iris()
parameters = {'kernel':('linear', 'rbf'), 'C':[1, 10]}
clf = model_selection.GridSearchCV(svm.SVC(), parameters)
clf.fit(iris.data, iris.target)
estimator = clf.best_estimator_ctrl + cfrom sklearn importimport module from scikit-learn | load_irisloads Iris dataset |
parametersparameters dictionary to run grid search accross | .GridSearchCV(creates GridSearchCV model |
svm.SVC()use SVC model as an estimator | .fit(train transformation model |
.best_estimator_returns estimator which gave highest score | |
Usage example
from sklearn import svm, datasets, model_selection
iris = datasets.load_iris()
parameters = {'kernel':('linear', 'rbf'), 'C':[1, 10]}
clf = model_selection.GridSearchCV(svm.SVC(), parameters)
clf.fit(iris.data, iris.target)
print(clf.best_estimator_)output
SVC(C=1, kernel='linear')
Related
More of Python Scikit Learn
- Use imputer for one column only
- Grid search CV example
- How to get best parameters from grid search CV
- Using quantile regression
- How to inverse transform PCA reduced data
- PCA dimensionality reduction example
- Feature selection using KBest and chi2
- Using OneClassSVM for outliers detection example
- Adaboost regression example
See more codes...