决策树对鸢尾花数据的两特征组合的分类结果

特征:   花萼长度  +  花萼宽度 	预测正确数目: 123 	准确率: 82.00%
特征:   花萼长度  +  花瓣长度 	预测正确数目: 145 	准确率: 96.67%
特征:   花萼长度  +  花瓣宽度 	预测正确数目: 144 	准确率: 96.00%
特征:   花萼宽度  +  花瓣长度 	预测正确数目: 143 	准确率: 95.33%
特征:   花萼宽度  +  花瓣宽度 	预测正确数目: 145 	准确率: 96.67%
特征:   花瓣长度  +  花瓣宽度 	预测正确数目: 147 	准确率: 98.00%

实现代码

#coding:utf-8

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
from sklearn.tree import DecisionTreeClassifier

def iris_type(s):
    it={'Iris-setosa': 0, 'Iris-versicolor': 1, 'Iris-virginica': 2}
    return it[s]

iris_feature = u'花萼长度', u'花萼宽度', u'花瓣长度', u'花瓣宽度'

if __name__ == "__main__":
    mpl.rcParams['font.sans-serif']=[u'SimHei'] #黑体 FangSong/KaiTi
    mpl.rcParams['axes.unicode_minus']=False

    path = '10.iris.data'
    #加载数据   converters={4:iris_type} 将第四列数据替换成iris_type函数返回的值
    data=np.loadtxt(path,dtype=float,delimiter=',',converters={4:iris_type})
    #前4列是特征向量 赋给x_prime,最后一列标签列赋给y
    x_prime,y=np.split(data,(4,),axis=1)

    # print 'x_prime'
    # print x_prime
    # print 'y='
    # print y
    # print '----------------'

    #(0,1)表示取第0列和
  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值