数据处理
# -*- coding: utf-8 -*-
"""
Created on Mon Apr 24 16:29:07 2017
@author: Administrator
"""
import pandas as pd
inputfile = 'F:/data/chapter9/demo/data/consider.xls' #数据文件
outputfile= 'F:/data/chapter9/demo/data/consider_o.xls'
data=pd.read_excel(inputfile)
'''
for i in range(data['空气等级'].size):
if data['空气等级'][i]=='I':
data['空气等级'][i]=1
elif data['空气等级'][i]=='II':
data['空气等级'][i]=2
elif data['空气等级'][i]=='III':
data['空气等级'][i]=3
elif data['空气等级'][i]=='IV':
data['空气等级'][i]=4
elif data['空气等级'][i]=='V':
data['空气等级'][i]=5
elif data['空气等级'][i]=='VI':
data['空气等级'][i]=6
elif data['空气等级'][i]=='VII':
data['空气等级'][i]=7
'''
data['空气等级'][data['空气等级']=='I']=1
data['空气等级'][data['空气等级']=='II']=2
data['空气等级'][data['空气等级']=='III']=3
data['空气等级'][data['空气等级']=='IV']=4
data['空气等级'][data['空气等级']=='V']=5
data['空气等级'][data['空气等级']=='VI']=6
data['空气等级'][data['空气等级']=='VII']=7
data.to_excel(outputfile)
模型训练
# -*- coding: utf-8 -*-
"""
Created on Mon Apr 24 16:58:23 2017
@author: Administrator
"""
#-*- coding: utf-8 -*-
import pandas as pd
from sklearn.tree import DecisionTreeClassifier as DTC
inputfile = 'F:/data/chapter9/demo/data/consider_o.xls' #数据文件
outputfile1 = 'F:/data/chapter9/demo/tmp/train_consider.xls' #训练样本混淆矩阵保存路径
outputfile2 = 'F:/data/chapter9/demo/tmp/test_consider.xls' #测试样本混淆矩阵保存路径
data = pd.read_excel(inputfile) #读取数据,指定编码为gbk
data = data.as_matrix()
from numpy.random import shuffle #引入随机函数
shuffle(data) #随机打乱数据
data_train = data[:int(0.8*len(data)), :] #选取前80%为训练数据
data_test = data[int(0.8*len(data)):, :] #选取前20%为测试数据
#构造特征和标签
x_train = data_train[:, 0:5]*30
y_train = data_train[:, 6].astype(int)
x_test = data_test[:, 0:5]*30
y_test = data_test[:, 6].astype(int)
model=DTC(criterion='entropy')
model.fit(x_train,y_train)
print(model.score(x_test,y_test))
#导入输出相关的库,生成混淆矩阵
from sklearn import metrics
cm_train = metrics.confusion_matrix(y_train, model.predict(x_train)) #训练样本的混淆矩阵
cm_test = metrics.confusion_matrix(y_test, model.predict(x_test)) #测试样本的混淆矩阵
#保存结果
pd.DataFrame(cm_train, index = range(1, 8), columns = range(1, 8)).to_excel(outputfile1)
pd.DataFrame(cm_test, index = range(1, 8), columns = range(1, 8)).to_excel(outputfile2)
print(model.score(x_train,y_train))