Python决策树鸢尾花类别分类(1)，头条面试题

#决策树 鸢尾花类别

import sklearn.datasets as dSets

import pandas as pd;

import matplotlib.pyplot as plt

from sklearn.model_selection import train_test_split

import sklearn.tree as sklTree

获取数据

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”)

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