本文对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)