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

0 vote131 views1 answer
Invalid literal for int() with base 10 python
Python
6 votes134 views5 answers
No module named env.__main__; 'env' is a package and cannot be directly executed
Python
0 vote108 views1 answer
Count wit for loop pthoon
Python
0 vote119 views1 answer
How to check nth prime in python
Python
0 vote110 views1 answer
MEMORY MANAGEMENT SYSTEM IN PYTHON
Python
22 votes112 views1 answer
Python test if number in string
Python
33 votes113 views6 answers
How to create a virtual environment in anaconda
Python
0 vote70 views2 answers
Nested renamer is not supported pandas
Python
10 votes113 views2 answers
Pygame get mouse position
Python
0 vote122 views1 answer
Python set timezone windows
Python