分类决策树

本文对sklearn中的分类决策树做大致介绍,详细内容参见sklearn官网

在sklearn中,分类决策树是一个名为sklearn.tree.DecisionTreeClassifier的类,它的构造函数的参数包括

criterion:不纯度指标,可选gini或entropy

splitter:分支策略,选best为最佳分支,选random为最佳随机分支

max_depth:树的最大深度

最简单的调用代码:

clf = DecisionTreeClassifier() 
clf = clf.fit(Xtrain, Ytrain) 
score = clf.score(Xtest, Ytest)  
原文地址:https://www.cnblogs.com/bill-h/p/13279193.html