在审核决策树文档here时,我注意到分类器没有办法调整拟合的“顺序”。具体来说,关于电话:
tree.DecisionTreeClassifier()
我想玩高/低“订单”来看看决策表面的视觉变化。
对 Regressor 的调用似乎确实具有以下功能:
regr_1 = DecisionTreeRegressor(max_depth=2)
regr_2 = DecisionTreeRegressor(max_depth=5)
DecisionTreeClassifier()
调用没有可比较的参数吗?在某些情况下,我认为这是至关重要的。
答案 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)