Voting classifier grid search

from sklearn.linear_model import LogisticRegression
from sklearn.svm import SVC
from sklearn.ensemble import VotingClassifier
from sklearn.model_selection import GridSearchCV

X = np.array([[-1.0, -1.0], [-1.2, -1.4], [-3.4, -2.2], [1.1, 1.2],[-1.0, -1.0], [-1.2, -1.4], [-3.4, -2.2], [1.1, 1.2]])
y = np.array([1, 1, 2, 2,1, 1, 2, 2])

eclf = VotingClassifier(estimators=[ 
    ('svm', SVC(probability=True)),
    ('lr', LogisticRegression()),
    ], voting='soft')

#Use the key for the classifier followed by __ and the attribute
params = {'lr__C': [1.0, 100.0],
      'svm__C': [2,3,4],}

grid = GridSearchCV(estimator=eclf, param_grid=params, cv=2)

grid.fit(X,y)

print (grid.best_params_)
#{'lr__C': 1.0, 'svm__C': 2}

Source: stackoverflow.com

Similar codes

5 votes177 views1 answer
Check if special character in string python
Python
0 vote181 views1 answer
Use a callable instead, e.g., use `dict` instead of `{}`
Python
13 votes180 views3 answers
Find the factorial of a given integer in python
Python
6 votes177 views2 answers
Python selenium save cookies
Python
0 vote176 views1 answer
Pillow rgb to grayscale
Python
21 votes198 views6 answers
Last element of list python
Python
14 votes177 views5 answers
Python while continue
Python
155 votes268 views6 answers
File exist python
Python
20 votes218 views4 answers
Button tkinter
Python
2 votes184 views4 answers
Decision tree classifier sklearn
Python