python利用sklearn包编写决策树源代码
2017-12-21 16:07
1026 查看
本文实例为大家分享了python编写决策树源代码,供大家参考,具体内容如下
因为最近实习的需要,所以用python里的sklearn包重新写了一次决策树。
工具:sklearn,将dot文件转化为pdf格式(是为了将形成的决策树可视化)graphviz-2.38,下载解压之后将其中的bin文件的目录添加进环境变量
源代码如下:
from sklearn.feature_extraction import DictVectorizer import csv from sklearn import tree from sklearn import preprocessing from sklearn.externals.six import StringIO from xml.sax.handler import feature_external_ges from numpy.distutils.fcompiler import dummy_fortran_file # Read in the csv file and put features into list of dict and list of class label allElectronicsData = open(r'E:/DeepLearning/resources/AllElectronics.csv', 'rt') reader = csv.reader(allElectronicsData) headers = next(reader) featureList = [] lableList = [] for row in reader: lableList.append(row[len(row)-1]) rowDict = {} #不包括len(row)-1 for i in range(1,len(row)-1): rowDict[headers[i]] = row[i] featureList.append(rowDict) print(featureList) vec = DictVectorizer() dummX = vec.fit_transform(featureList).toarray() print(str(dummX)) lb = preprocessing.LabelBinarizer() dummY = lb.fit_transform(lableList) print(str(dummY)) #entropy=>ID3 clf = tree.DecisionTreeClassifier(criterion='entropy') clf = clf.fit(dummX, dummY) print("clf:"+str(clf)) #可视化tree with open("resultTree.dot",'w')as f: f = tree.export_graphviz(clf, feature_names=vec.get_feature_names(),out_file = f) #对于新的数据怎样来查看它的分类 oneRowX = dummX[0,:] print("oneRowX: "+str(oneRowX)) newRowX = oneRowX newRowX[0] = 1 newRowX[2] = 0 predictedY = clf.predict(newRowX) print("predictedY: "+ str(predictedY))
这里的AllElectronics.csv,形式如下图所示:
今天早上好不容易将jdk、eclipse以及pydev装进linux,但是,但是,但是,想装numpy的时候,总是报错,发现是没有gcc,然后又去装gcc,真是醉了,到现在gcc还是没有装成功,再想想方法
您可能感兴趣的文章:
相关文章推荐
- 利用aardio给python编写图形界面
- 利用Python编写Web自动化测试
- 利用python 多进程编写的简单实例
- 利用Python编写一个程序 统计当前目录下的每个文件类型的文件数目
- 利用C++ Boost编写扩展Python模块
- 【Grades Crawler】利用python编写爬虫 爬取西电教务处成绩并本地保存
- Python - 利用python编写的memcached启动脚本
- 利用Python的SocketServer框架编写网络服务程序
- 利用Python编写linux自动备份脚本
- QGis 利用Python Console编写脚本进行批量处理
- 项目实战_Python.利用Pygame编写微信打飞机小游戏完整系列?
- 利用python 编写 简单 虚拟机管理
- Python利用hadoop Streaming编写的Map-Reduce程序命令运行和本地调试运行
- Python利用unittest框架编写接口测试小栗子
- 机器学习-python利用SVD编写推荐引擎
- 利用C++ Boost编写扩展Python模块
- 利用python编写设计多线程web服务器(计算机网络_自顶向下第六版_第二章1和4的编程作业)
- 利用python编写东北大学自动联网程序
- 利用python aiohttp编写web框架
- 利用python基础编写简单的日历