基于决策树的分类算法
初始化模型
分成测试集和训练集
模型拟合
模型预测
模型评估
模型评估
#选择其类别为0和1的样本 (不包括类别为2的样本)
data_target_part = data[data[‘Species’].isin([0,1])][[‘Species’]]
data_features_part = data[data[‘Species’].isin([0,1])][[‘Culmen Length (mm)’,‘Culmen Depth (mm)’,
‘Flipper Length (mm)’,‘Body Mass (g)’]]