SKLearn决策树分类器深度/顺序

时间:2017-05-06 20:01:21

标签: python python-3.x scikit-learn

在审核决策树文档here时,我注意到分类器没有办法调整拟合的“顺序”。具体来说,关于电话:

tree.DecisionTreeClassifier()

我想玩高/低“订单”来看看决策表面的视觉变化。

Regressor 的调用似乎确实具有以下功能:

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

DecisionTreeClassifier()调用没有可比较的参数吗?在某些情况下,我认为这是至关重要的。

1 个答案:

答案 0 :(得分: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)