Python数据分析(9)— 用决策树进行分类

在上一篇博文Python数据分析(8)—-用python实现数据分层抽样中,实现了实验数据的抽取,那么在本文中,将用上述抽取到的数据进行实验,也就是用决策树进行分类。 在讲解实际的决策树分类之前,需要介绍一下决策树分类的sklearn中决策树模型参数释义: ''' scikit-learn中有两类决策树,它们均采用优化的CART决策树算法。 (1)回归决策树:DecisionTreeRegressor() (2)分类决策树:DecisionTreeClassifier() ''' from sklearn.tree import DecisionTreeRegressor ''' 回归决策树