当前位置:   article > 正文

Python决策树鸢尾花类别分类(1)_构建一个决策树对鸢尾花数据集(iris)进行分类,描述主要过程并粘贴实现代码

构建一个决策树对鸢尾花数据集(iris)进行分类,描述主要过程并粘贴实现代码

获取数据

iris = dSets.load_iris();

#print(iris);

data = iris.data;

target = iris.target;

print(“data”,data); #属性值

print(“target”,target); #结果

输出

在这里插入图片描述

在这里插入图片描述

划分测试集和训练集和

#画分测试机和训练集 训练测试7:3

x_train,x_test,y_train,y_test = train_test_split(data,target,test_size=0.3,random_state=1);

决策树参数说明

1.criterion:string, optional (default=“mse”)

它指定了切分质量的评价准则。默认为’mse’(mean squared error)。

2.splitter:string, optional (default=“best”)

它指定了在每个节点切分的策略。有两种切分策咯:

(1).splitter=‘best’:表示选择最优的切分特征和切分点。

(2).split

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/701819
推荐阅读
相关标签
  

闽ICP备14008679号