2017-05-06 29 views
0

デシジョンツリーのドキュメントhereを見て、分類器には適合の「順序」を調整する手段がないことに気付きました。具体的には、電話に関して:SKLearnデシジョンツリー分類器の深さ/順序

tree.DecisionTreeClassifier() 

決定表の視界がどのように変化するかを見るために、高低の「注文」で遊びたいと思います。

REGRESSORへの呼び出しは、この機能を持っているように見えるん:

regr_1 = DecisionTreeRegressor(max_depth=2) 
regr_2 = DecisionTreeRegressor(max_depth=5) 

DecisionTreeClassifier()呼び出しが同等の引数を持っていませんか?私はいくつかの例でそれが不可欠であると推測するでしょう。

答えて

1

DecisionTreeClassifierも引数がmax_depthです。 the docsを参照してください。

>>> from sklearn.datasets import load_iris 
>>> from sklearn.model_selection import cross_val_score 
>>> from sklearn.tree import DecisionTreeClassifier 
>>> clf = DecisionTreeClassifier(max_depth=2) 
>>> iris = load_iris() 
>>> cross_val_score(clf, iris.data, iris.target, cv=10) 
関連する問題