grid_search
授業の進め方が分からなくて困っています。max_depth
で使用できる最適なパラメータを見つけたいですRandomForestClassifier
。検索を実行したい可能なオプションを指定し、モジュールが「最適な」max_depth
オプションを出力することを期待しました。
from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestClassifier
from sklearn import grid_search
iris= load_iris()
forest_parameters = {'max_depth': [1,2,3,4]}
forest = RandomForestClassifier()
explorer = grid_search.GridSearchCV(forest, forest_parameters)
explorer.fit(iris['data'], iris['target'])
可能なオプションのセットが与えられた場合、explorer
グリッド検索モジュールが最適なパラメーターで返されることを期待していました。デフォルト値がまだ使用されているのはなぜですか? 「最適な」パラメーターを見つけるためにどのように使用できますか?max_depth
[1,2,3,4]
None
grid_search
Out[13]:
GridSearchCV(cv=None, error_score='raise',
estimator=RandomForestClassifier(bootstrap=True, class_weight=None, criterion='gini',
---> max_depth=None, max_features='auto', max_leaf_nodes=None,
min_samples_leaf=1, min_samples_split=2,
min_weight_fraction_leaf=0.0, n_estimators=10, n_jobs=1,
oob_score=False, random_state=None, verbose=0,
warm_start=False),
fit_params={}, iid=True, loss_func=None, n_jobs=1,
param_grid={'max_depth': [1, 2, 3, 4]}, pre_dispatch='2*n_jobs',
refit=True, score_func=None, scoring=None, verbose=0)