我们可以选择在sklearn中使用什么决策树算法吗?

时间:2015-12-11 18:20:30

标签: scikit-learn decision-tree

我的问题是我们可以选择在sklearn中使用什么决策树算法吗?

在sklearn的用户指南中,它提到了使用CART算法的优化版本。

我们可以更改为其他算法,例如C4.5吗?

2 个答案:

答案 0 :(得分:4)

没有。见documentation

scikit-learn uses an optimised version of the CART algorithm.

答案 1 :(得分:0)

但是有一个params标准我们可以选择使用“gini”或“entropy”:

clf = tree.DecisionTreeClassifier(criterion="entropy")
  

标准:字符串,可选(默认=“gini”)要测量的函数   分裂的质量。支持的标准是Gini的“gini”   杂质和信息增益的“熵”。

请参阅Docs