def guess_three(self, season, round_num):
feature = Feature(4)
feature.prepare_team_ranking()
with codecs.open('input/yingchao_result.csv', 'rb', 'utf-8') as file:
next(file)
for row in file:
row = row.strip('\r\n')
columns = row.split(',')
if season == int(columns[0]) and round_num == int(
columns[2]) and int(columns[1]) == 4:
home_team = Feature.team_name_clear(columns[3])
guest_team = Feature.team_name_clear(columns[4])
home_key = Feature.create_team_id('2017', columns[1],
home_team)
guest_key = Feature.create_team_id('2017', columns[1],
guest_team)
home_ranking = feature.get_team_ranking(home_key)
guest_ranking = feature.get_team_ranking(guest_key)
diff_ranking = home_ranking - guest_ranking
print home_team + '\t' + guest_team + '\t' + str(
home_ranking) + '\t' + str(guest_ranking) + '\t' + str(
diff_ranking)
def guess_three(self, season, round_num):
feature = Feature(38)
feature.prepare_team_ranking()
with codecs.open('input/meiguodalianmen_result.csv', 'rb',
'utf-8') as file:
next(file)
for row in file:
row = row.strip('\r\n')
columns = row.split(',')
if season == int(columns[0]) and round_num == int(
columns[2]) and int(columns[1]) == 38:
home_team = Feature.team_name_clear(columns[3])
guest_team = Feature.team_name_clear(columns[4])
home_key = Feature.create_team_id(columns[0], columns[1],
home_team)
guest_key = Feature.create_team_id(columns[0], columns[1],
guest_team)
home_ranking = feature.get_team_ranking(home_key)
guest_ranking = feature.get_team_ranking(guest_key)
ranking_diff = home_ranking - guest_ranking
if ranking_diff in self.LOST_EXP:
three = -1
elif ranking_diff <= self.WIN_LINE:
three = 1
elif ranking_diff >= self.LOST_LINE:
three = -1
else:
three = 0
print home_team + '\t' + guest_team + '\t' + str(
home_ranking) + '\t' + str(guest_ranking) + '\t' + str(
three)
