1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
import json, dateutil
from sqlalchemy import func
from itertools import combinations
from os import path
import __main__
from f1elo.model import *
class Elo:
def __init__(self, session):
self.session = session
self.config = json.load(
open(path.join(path.dirname(__main__.__file__), 'config', 'elo.json')))
def get_ranking(self, driver, rank_date=None):
rank = driver.get_ranking(rank_date)
if rank:
return rank.ranking
return self.config['initial_ranking']
def get_entry_ranking(self, entry, date=None):
return sum([self.get_ranking(d, date) for d in entry.drivers]) / len(entry.drivers)
def rank_race(self, race):
recent_date = race.date - dateutil.relativedelta.relativedelta(months=9)
recent_ratings = self.session.query(
func.min(Ranking.ranking).label('min'),
func.max(Ranking.ranking).label('max')
).filter(
Ranking.rank_date >= recent_date
).group_by(
Ranking._driver
)
changes_query = self.session.query(
func.avg(
func.abs(
recent_ratings.subquery().columns.min - recent_ratings.subquery().columns.max
)
)
)
recent_rank_changes = changes_query.scalar()
race_disparity = self.config['disparity']
if recent_rank_changes:
race_disparity -= recent_rank_changes
print(race_disparity, str(race.date))
entries = race.entries
entries_to_compare = []
rankings = {}
new_rankings = {}
for e in entries:
rankings[e] = self.get_entry_ranking(e, race.date)
new_rankings[e] = 0.0
if e.result_group:
entries_to_compare.append(e)
for c in combinations(entries_to_compare, 2):
score = self.get_score(
rankings[c[0]] - rankings[c[1]],
self.get_outcome(c),
self.get_importance(race,
[rankings[c[0]],
rankings[c[1]]]),
race_disparity
)
new_rankings[c[0]] += score
new_rankings[c[1]] -= score
return new_rankings
def get_importance(self, race, rankings):
base_importance = self.config['importance'][race.type.code]
min_rank = min(rankings)
if min_rank < min(self.config['importance_threshold']):
return base_importance
if min_rank <= max(self.config['importance_threshold']):
return base_importance * 0.75
return base_importance / 2
def get_outcome(self, entries):
if entries[0].result_group < entries[1].result_group:
return 1
elif entries[0].result_group > entries[1].result_group:
return 0
return 0.5
def get_score(self, difference, outcome, importance, disparity):
return importance * (outcome - 1 / (1 + (10 ** (-difference / disparity))))
|
