sklearn实现决策树算法
1、决策树算法是一种非参数的决策算法,它根据数据的不同特征进行多层次的分类和判断,最终决策出所需要预测的结果。 它既可以解决分类算法,也可以解决回归问题 ,具有很好的解释能力。另外,对于决策树的构建方法具有多种出发点,它具有多种构建方式,如何构建决策树的出发点主要在于决策树每一个决策点上需要 在哪些维度上进行划分以及在这些维度的哪些阈值节点做划分 等细节问题。 具体在sklearn中调用决策树算法解决分类问题和回归问题的程序代码如下所示: #1-1导入基础训练数据集import numpy as npimport matplotlib.pyplot as pltfrom sklearn import datasetsd=datasets.load_iris()x=d.data[:,2:]y=d.targetplt.figure()plt.scatter(x[y==0,0],x[y==0,1],color="r")plt.scatter(x[y==1,0],x[y==1,1],color="g")plt.scatter(x[y==2,0],x[y==2,1],color="b")plt.show()#1-2导入sklearn中的决策树算法进行数据的分类问题实现训练预测from sklearn.tree import DecisionTreeClassifierdt1