分类决策树

本文对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)  
posted @ 2020-07-10 14:49  Bill_H  阅读(149)  评论(0编辑  收藏  举报