python+决策树1

#pandas库(Python Data Analysis的简写)是用来加载、管理和处理数据的。它在后台处理数据结构,支持计算均值等分析方法。 
import pandas as pd
from sklearn.model_selection._validation import cross_val_score   #交叉检验,计算平均正确率
from sklearn.tree import DecisionTreeClassifier
import numpy as np
clf = DecisionTreeClassifier(random_state=14)

filename = "dicision_trees_sample.csv"
#修复参数,将Date列的值由字符串改为日期类型。
dataset = pd.read_csv(filename, parse_dates=["Date"])  
#定义表头即定义属性列。 
dataset.columns = ["Date","StartTime","VistorTeam","VisitorPTS","HomeTeam","HomePTS","ScoreType","OT?","Notes"]

#添加新特征,主场获胜与否(1表示主场获胜,0表示主场未获胜),作为预测的结果是否正确的标准。
dataset["HomeWin"] = dataset["VisitorPTS"] < dataset["HomePTS"]
x_c = dataset["HomeWin"].values
from collections import defaultdict
won_last = defaultdict(int)

#dataset["HomeLastWin"]和dataset["VistorLastWin"]必须要定义,否则下面循环里的row赋值将没有作用。
#球队上场的输赢作为判断依据(思路是这样的:如果上一次比赛球队A获胜,则预测下一场比赛A也会获胜)
dataset["HomeLastWin"] = 0     #默认每一个主场球队在第一次出现的时候都是输的
dataset["VistorLastWin"] = 0   #默认每一个客场球队在第一次出现的时候都是输的
#dataset.sort_values("Date"),如果报错,可以将sort_values改为sort(pandas版本问题),数据按照时间排序,因为上一次比赛与下一场比赛是通过时间进行区分的。
for index, row in dataset.sort_values("Date").iterrows():
    homeTeam = row["HomeTeam"]
    visitorTeam = row["VistorTeam"]
    row["HomeLastWin"] = won_last[homeTeam]
    row["VistorLastWin"] = won_last[visitorTeam]
    dataset.ix[index] = row
    won_last[homeTeam] = row["HomeWin"]      
    won_last[visitorTeam] = not row["HomeWin"] 

#属性列HomeLastWin和VistorLastWin的取值作为预测球队下场是否能获胜的依据
x_previouswin = dataset[["HomeLastWin", "VistorLastWin"]].values
scores = cross_val_score(clf, x_previouswin, x_c, scoring="accuracy")
print("使用上场比赛的结果预测下场比赛,Accuracy: {0:.1f}%".format(np.mean(scores) * 100)) 

#创建新特征,提高准确率,球队排名作为预测是否获胜的依据
dataset["HomeTeamRanksHiger"] = 0
standings_filename = "leagues_NBA_2013_standings_expanded-standings.csv"
standings = pd.read_csv(standings_filename)
for index, row in dataset.sort_values("Date").iterrows():
    homeTeam = row["HomeTeam"]
    visitorTeam = row["VistorTeam"]
    #处理有些球队更名问题
    if homeTeam == "New Orleans Pelicans":
        homeTeam = "New Orleans Hornets"
    elif visitorTeam == "New Orleans Pelicans":
        visitorTeam = "New Orleans Hornets"
    #standings[ standings["Team"]== homeTeam ],首先在standings筛选出homeTeam,然后得到它的排名
    homeRank = standings[ standings["Team"]== homeTeam ]["Rk"].values[0]   #存放主场球队排名
    visitorRank = standings[ standings["Team"]== visitorTeam]["Rk"].values[0]  #存放客场球队排名
    row["HomeTeamRanksHiger"] = int(homeRank > visitorRank)
    dataset.ix[index] = row
x_homehigher = dataset[["HomeLastWin", "VistorLastWin","HomeTeamRanksHiger"]].values
scores = cross_val_score(clf, x_homehigher, x_c, scoring="accuracy")
print("使用上场比赛的结果+球队排名预测下场比赛,Accuracy: {0:.1f}%".format(np.mean(scores) * 100)) 

运行结果截图:
这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值