SKLearn決策樹分類深度/訂購

Question

在回顧決策樹文檔here，我注意到分類不必調整配合的「秩序」的一種手段。具體來說，關於電話：

tree.DecisionTreeClassifier() 

我想玩高/低「命令」，看看如何決定表面視覺變化。

到迴歸調用似乎有這樣的特徵：

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

是否DecisionTreeClassifier()電話不具備可比性的論點？我會在某些情況下，假定這將是至關重要的。

Answer 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)