1from sklearn.model_selection import GridSearchCV
2from sklearn.tree import DecisionTreeClassifier
3from sklearn.datasets import make_classification
4
5X, y = make_classification(n_samples = 100, n_features = 3,
6n_informative = 2, n_redundant = 1)
7
8param_grid = [{'max_depth': [2, 3, 4, 5, 6], 'min_samples_split': [2, 3, 4, 5, 6]},
9 {'min_samples_leaf': [2, 3, 4, 5, 6]},
10 {'min_impurity_decrease': [0.2, 0.3, 0.4, 0.5, 0.6],
11 'ccp_alpha': [0.1, 0.2, 0.3, 0.4, 0.5, 0.6]}]
12
13gscv = GridSearchCV(DecisionTreeClassifier(), param_grid, cv = 3)
14gscv.fit(X, y)
15
16print(gscv.best_params_)