From 400b5bb81e7435da8637446405d06db18c6d120d Mon Sep 17 00:00:00 2001 From: art Date: Fri, 4 Oct 2019 09:02:46 -0500 Subject: [PATCH] upload trueskill for testing purposes --- data analysis/analysis/trueskill/__about__.py | 11 + data analysis/analysis/trueskill/__init__.py | 726 ++++++++++++++++++ .../__pycache__/__about__.cpython-37.pyc | Bin 0 -> 414 bytes .../__pycache__/__init__.cpython-37.pyc | Bin 0 -> 25214 bytes .../__pycache__/backends.cpython-37.pyc | Bin 0 -> 4073 bytes .../__pycache__/deprecated.cpython-37.pyc | Bin 0 -> 4108 bytes .../__pycache__/factorgraph.cpython-37.pyc | Bin 0 -> 7279 bytes .../__pycache__/mathematics.cpython-37.pyc | Bin 0 -> 9250 bytes data analysis/analysis/trueskill/backends.py | 135 ++++ .../analysis/trueskill/deprecated.py | 134 ++++ .../analysis/trueskill/factorgraph.py | 199 +++++ .../analysis/trueskill/mathematics.py | 260 +++++++ 12 files changed, 1465 insertions(+) create mode 100644 data analysis/analysis/trueskill/__about__.py create mode 100644 data analysis/analysis/trueskill/__init__.py create mode 100644 data analysis/analysis/trueskill/__pycache__/__about__.cpython-37.pyc create mode 100644 data analysis/analysis/trueskill/__pycache__/__init__.cpython-37.pyc create mode 100644 data analysis/analysis/trueskill/__pycache__/backends.cpython-37.pyc create mode 100644 data analysis/analysis/trueskill/__pycache__/deprecated.cpython-37.pyc create mode 100644 data analysis/analysis/trueskill/__pycache__/factorgraph.cpython-37.pyc create mode 100644 data analysis/analysis/trueskill/__pycache__/mathematics.cpython-37.pyc create mode 100644 data analysis/analysis/trueskill/backends.py create mode 100644 data analysis/analysis/trueskill/deprecated.py create mode 100644 data analysis/analysis/trueskill/factorgraph.py create mode 100644 data analysis/analysis/trueskill/mathematics.py diff --git a/data analysis/analysis/trueskill/__about__.py b/data analysis/analysis/trueskill/__about__.py new file mode 100644 index 00000000..8e5e3ee3 --- /dev/null +++ b/data analysis/analysis/trueskill/__about__.py @@ -0,0 +1,11 @@ +# -*- coding: utf-8 -*- +""" + trueskill.__about__ + ~~~~~~~~~~~~~~~~~~~ +""" +__version__ = '0.4.5' +__license__ = 'BSD' +__author__ = 'Heungsub Lee' +__author_email__ = 'sub@subl.ee' +__url__ = 'http://trueskill.org/' +__description__ = 'The video game rating system' diff --git a/data analysis/analysis/trueskill/__init__.py b/data analysis/analysis/trueskill/__init__.py new file mode 100644 index 00000000..ac5facd6 --- /dev/null +++ b/data analysis/analysis/trueskill/__init__.py @@ -0,0 +1,726 @@ +# -*- coding: utf-8 -*- +""" + trueskill + ~~~~~~~~~ + + The video game rating system. + + :copyright: (c) 2012-2016 by Heungsub Lee + :license: BSD, see LICENSE for more details. + +""" +from __future__ import absolute_import + +from itertools import chain +import math + +from six import iteritems +from six.moves import map, range, zip + +from .__about__ import __version__ # noqa +from .backends import choose_backend +from .factorgraph import (LikelihoodFactor, PriorFactor, SumFactor, + TruncateFactor, Variable) +from .mathematics import Gaussian, Matrix + + +__all__ = [ + # TrueSkill objects + 'TrueSkill', 'Rating', + # functions for the global environment + 'rate', 'quality', 'rate_1vs1', 'quality_1vs1', 'expose', 'setup', + 'global_env', + # default values + 'MU', 'SIGMA', 'BETA', 'TAU', 'DRAW_PROBABILITY', + # draw probability helpers + 'calc_draw_probability', 'calc_draw_margin', + # deprecated features + 'transform_ratings', 'match_quality', 'dynamic_draw_probability', +] + + +#: Default initial mean of ratings. +MU = 25. +#: Default initial standard deviation of ratings. +SIGMA = MU / 3 +#: Default distance that guarantees about 76% chance of winning. +BETA = SIGMA / 2 +#: Default dynamic factor. +TAU = SIGMA / 100 +#: Default draw probability of the game. +DRAW_PROBABILITY = .10 +#: A basis to check reliability of the result. +DELTA = 0.0001 + + +def calc_draw_probability(draw_margin, size, env=None): + """Calculates a draw-probability from the given ``draw_margin``. + + :param draw_margin: the draw-margin. + :param size: the number of players in two comparing teams. + :param env: the :class:`TrueSkill` object. Defaults to the global + environment. + + """ + if env is None: + env = global_env() + return 2 * env.cdf(draw_margin / (math.sqrt(size) * env.beta)) - 1 + + +def calc_draw_margin(draw_probability, size, env=None): + """Calculates a draw-margin from the given ``draw_probability``. + + :param draw_probability: the draw-probability. + :param size: the number of players in two comparing teams. + :param env: the :class:`TrueSkill` object. Defaults to the global + environment. + + """ + if env is None: + env = global_env() + return env.ppf((draw_probability + 1) / 2.) * math.sqrt(size) * env.beta + + +def _team_sizes(rating_groups): + """Makes a size map of each teams.""" + team_sizes = [0] + for group in rating_groups: + team_sizes.append(len(group) + team_sizes[-1]) + del team_sizes[0] + return team_sizes + + +def _floating_point_error(env): + if env.backend == 'mpmath': + msg = 'Set "mpmath.mp.dps" to higher' + else: + msg = 'Cannot calculate correctly, set backend to "mpmath"' + return FloatingPointError(msg) + + +class Rating(Gaussian): + """Represents a player's skill as Gaussian distrubution. + + The default mu and sigma value follows the global environment's settings. + If you don't want to use the global, use :meth:`TrueSkill.create_rating` to + create the rating object. + + :param mu: the mean. + :param sigma: the standard deviation. + + """ + + def __init__(self, mu=None, sigma=None): + if isinstance(mu, tuple): + mu, sigma = mu + elif isinstance(mu, Gaussian): + mu, sigma = mu.mu, mu.sigma + if mu is None: + mu = global_env().mu + if sigma is None: + sigma = global_env().sigma + super(Rating, self).__init__(mu, sigma) + + def __int__(self): + return int(self.mu) + + def __long__(self): + return long(self.mu) + + def __float__(self): + return float(self.mu) + + def __iter__(self): + return iter((self.mu, self.sigma)) + + def __repr__(self): + c = type(self) + args = ('.'.join([c.__module__, c.__name__]), self.mu, self.sigma) + return '%s(mu=%.3f, sigma=%.3f)' % args + + +class TrueSkill(object): + """Implements a TrueSkill environment. An environment could have + customized constants. Every games have not same design and may need to + customize TrueSkill constants. + + For example, 60% of matches in your game have finished as draw then you + should set ``draw_probability`` to 0.60:: + + env = TrueSkill(draw_probability=0.60) + + For more details of the constants, see `The Math Behind TrueSkill`_ by + Jeff Moser. + + .. _The Math Behind TrueSkill:: http://bit.ly/trueskill-math + + :param mu: the initial mean of ratings. + :param sigma: the initial standard deviation of ratings. The recommended + value is a third of ``mu``. + :param beta: the distance which guarantees about 76% chance of winning. + The recommended value is a half of ``sigma``. + :param tau: the dynamic factor which restrains a fixation of rating. The + recommended value is ``sigma`` per cent. + :param draw_probability: the draw probability between two teams. It can be + a ``float`` or function which returns a ``float`` + by the given two rating (team performance) + arguments and the beta value. If it is a + ``float``, the game has fixed draw probability. + Otherwise, the draw probability will be decided + dynamically per each match. + :param backend: the name of a backend which implements cdf, pdf, ppf. See + :mod:`trueskill.backends` for more details. Defaults to + ``None``. + + """ + + def __init__(self, mu=MU, sigma=SIGMA, beta=BETA, tau=TAU, + draw_probability=DRAW_PROBABILITY, backend=None): + self.mu = mu + self.sigma = sigma + self.beta = beta + self.tau = tau + self.draw_probability = draw_probability + self.backend = backend + if isinstance(backend, tuple): + self.cdf, self.pdf, self.ppf = backend + else: + self.cdf, self.pdf, self.ppf = choose_backend(backend) + + def create_rating(self, mu=None, sigma=None): + """Initializes new :class:`Rating` object, but it fixes default mu and + sigma to the environment's. + + >>> env = TrueSkill(mu=0, sigma=1) + >>> env.create_rating() + trueskill.Rating(mu=0.000, sigma=1.000) + + """ + if mu is None: + mu = self.mu + if sigma is None: + sigma = self.sigma + return Rating(mu, sigma) + + def v_win(self, diff, draw_margin): + """The non-draw version of "V" function. "V" calculates a variation of + a mean. + """ + x = diff - draw_margin + denom = self.cdf(x) + return (self.pdf(x) / denom) if denom else -x + + def v_draw(self, diff, draw_margin): + """The draw version of "V" function.""" + abs_diff = abs(diff) + a, b = draw_margin - abs_diff, -draw_margin - abs_diff + denom = self.cdf(a) - self.cdf(b) + numer = self.pdf(b) - self.pdf(a) + return ((numer / denom) if denom else a) * (-1 if diff < 0 else +1) + + def w_win(self, diff, draw_margin): + """The non-draw version of "W" function. "W" calculates a variation of + a standard deviation. + """ + x = diff - draw_margin + v = self.v_win(diff, draw_margin) + w = v * (v + x) + if 0 < w < 1: + return w + raise _floating_point_error(self) + + def w_draw(self, diff, draw_margin): + """The draw version of "W" function.""" + abs_diff = abs(diff) + a, b = draw_margin - abs_diff, -draw_margin - abs_diff + denom = self.cdf(a) - self.cdf(b) + if not denom: + raise _floating_point_error(self) + v = self.v_draw(abs_diff, draw_margin) + return (v ** 2) + (a * self.pdf(a) - b * self.pdf(b)) / denom + + def validate_rating_groups(self, rating_groups): + """Validates a ``rating_groups`` argument. It should contain more than + 2 groups and all groups must not be empty. + + >>> env = TrueSkill() + >>> env.validate_rating_groups([]) + Traceback (most recent call last): + ... + ValueError: need multiple rating groups + >>> env.validate_rating_groups([(Rating(),)]) + Traceback (most recent call last): + ... + ValueError: need multiple rating groups + >>> env.validate_rating_groups([(Rating(),), ()]) + Traceback (most recent call last): + ... + ValueError: each group must contain multiple ratings + >>> env.validate_rating_groups([(Rating(),), (Rating(),)]) + ... #doctest: +ELLIPSIS + [(truekill.Rating(...),), (trueskill.Rating(...),)] + + """ + # check group sizes + if len(rating_groups) < 2: + raise ValueError('Need multiple rating groups') + elif not all(rating_groups): + raise ValueError('Each group must contain multiple ratings') + # check group types + group_types = set(map(type, rating_groups)) + if len(group_types) != 1: + raise TypeError('All groups should be same type') + elif group_types.pop() is Rating: + raise TypeError('Rating cannot be a rating group') + # normalize rating_groups + if isinstance(rating_groups[0], dict): + dict_rating_groups = rating_groups + rating_groups = [] + keys = [] + for dict_rating_group in dict_rating_groups: + rating_group, key_group = [], [] + for key, rating in iteritems(dict_rating_group): + rating_group.append(rating) + key_group.append(key) + rating_groups.append(tuple(rating_group)) + keys.append(tuple(key_group)) + else: + rating_groups = list(rating_groups) + keys = None + return rating_groups, keys + + def validate_weights(self, weights, rating_groups, keys=None): + if weights is None: + weights = [(1,) * len(g) for g in rating_groups] + elif isinstance(weights, dict): + weights_dict, weights = weights, [] + for x, group in enumerate(rating_groups): + w = [] + weights.append(w) + for y, rating in enumerate(group): + if keys is not None: + y = keys[x][y] + w.append(weights_dict.get((x, y), 1)) + return weights + + def factor_graph_builders(self, rating_groups, ranks, weights): + """Makes nodes for the TrueSkill factor graph. + + Here's an example of a TrueSkill factor graph when 1 vs 2 vs 1 match:: + + rating_layer: O O O O (PriorFactor) + | | | | + | | | | + perf_layer: O O O O (LikelihoodFactor) + | \ / | + | | | + team_perf_layer: O O O (SumFactor) + \ / \ / + | | + team_diff_layer: O O (SumFactor) + | | + | | + trunc_layer: O O (TruncateFactor) + + """ + flatten_ratings = sum(map(tuple, rating_groups), ()) + flatten_weights = sum(map(tuple, weights), ()) + size = len(flatten_ratings) + group_size = len(rating_groups) + # create variables + rating_vars = [Variable() for x in range(size)] + perf_vars = [Variable() for x in range(size)] + team_perf_vars = [Variable() for x in range(group_size)] + team_diff_vars = [Variable() for x in range(group_size - 1)] + team_sizes = _team_sizes(rating_groups) + # layer builders + def build_rating_layer(): + for rating_var, rating in zip(rating_vars, flatten_ratings): + yield PriorFactor(rating_var, rating, self.tau) + def build_perf_layer(): + for rating_var, perf_var in zip(rating_vars, perf_vars): + yield LikelihoodFactor(rating_var, perf_var, self.beta ** 2) + def build_team_perf_layer(): + for team, team_perf_var in enumerate(team_perf_vars): + if team > 0: + start = team_sizes[team - 1] + else: + start = 0 + end = team_sizes[team] + child_perf_vars = perf_vars[start:end] + coeffs = flatten_weights[start:end] + yield SumFactor(team_perf_var, child_perf_vars, coeffs) + def build_team_diff_layer(): + for team, team_diff_var in enumerate(team_diff_vars): + yield SumFactor(team_diff_var, + team_perf_vars[team:team + 2], [+1, -1]) + def build_trunc_layer(): + for x, team_diff_var in enumerate(team_diff_vars): + if callable(self.draw_probability): + # dynamic draw probability + team_perf1, team_perf2 = team_perf_vars[x:x + 2] + args = (Rating(team_perf1), Rating(team_perf2), self) + draw_probability = self.draw_probability(*args) + else: + # static draw probability + draw_probability = self.draw_probability + size = sum(map(len, rating_groups[x:x + 2])) + draw_margin = calc_draw_margin(draw_probability, size, self) + if ranks[x] == ranks[x + 1]: # is a tie? + v_func, w_func = self.v_draw, self.w_draw + else: + v_func, w_func = self.v_win, self.w_win + yield TruncateFactor(team_diff_var, + v_func, w_func, draw_margin) + # build layers + return (build_rating_layer, build_perf_layer, build_team_perf_layer, + build_team_diff_layer, build_trunc_layer) + + def run_schedule(self, build_rating_layer, build_perf_layer, + build_team_perf_layer, build_team_diff_layer, + build_trunc_layer, min_delta=DELTA): + """Sends messages within every nodes of the factor graph until the + result is reliable. + """ + if min_delta <= 0: + raise ValueError('min_delta must be greater than 0') + layers = [] + def build(builders): + layers_built = [list(build()) for build in builders] + layers.extend(layers_built) + return layers_built + # gray arrows + layers_built = build([build_rating_layer, + build_perf_layer, + build_team_perf_layer]) + rating_layer, perf_layer, team_perf_layer = layers_built + for f in chain(*layers_built): + f.down() + # arrow #1, #2, #3 + team_diff_layer, trunc_layer = build([build_team_diff_layer, + build_trunc_layer]) + team_diff_len = len(team_diff_layer) + for x in range(10): + if team_diff_len == 1: + # only two teams + team_diff_layer[0].down() + delta = trunc_layer[0].up() + else: + # multiple teams + delta = 0 + for x in range(team_diff_len - 1): + team_diff_layer[x].down() + delta = max(delta, trunc_layer[x].up()) + team_diff_layer[x].up(1) # up to right variable + for x in range(team_diff_len - 1, 0, -1): + team_diff_layer[x].down() + delta = max(delta, trunc_layer[x].up()) + team_diff_layer[x].up(0) # up to left variable + # repeat until to small update + if delta <= min_delta: + break + # up both ends + team_diff_layer[0].up(0) + team_diff_layer[team_diff_len - 1].up(1) + # up the remainder of the black arrows + for f in team_perf_layer: + for x in range(len(f.vars) - 1): + f.up(x) + for f in perf_layer: + f.up() + return layers + + def rate(self, rating_groups, ranks=None, weights=None, min_delta=DELTA): + """Recalculates ratings by the ranking table:: + + env = TrueSkill() # uses default settings + # create ratings + r1 = env.create_rating(42.222) + r2 = env.create_rating(89.999) + # calculate new ratings + rating_groups = [(r1,), (r2,)] + rated_rating_groups = env.rate(rating_groups, ranks=[0, 1]) + # save new ratings + (r1,), (r2,) = rated_rating_groups + + ``rating_groups`` is a list of rating tuples or dictionaries that + represents each team of the match. You will get a result as same + structure as this argument. Rating dictionaries for this may be useful + to choose specific player's new rating:: + + # load players from the database + p1 = load_player_from_database('Arpad Emrick Elo') + p2 = load_player_from_database('Mark Glickman') + p3 = load_player_from_database('Heungsub Lee') + # calculate new ratings + rating_groups = [{p1: p1.rating, p2: p2.rating}, {p3: p3.rating}] + rated_rating_groups = env.rate(rating_groups, ranks=[0, 1]) + # save new ratings + for player in [p1, p2, p3]: + player.rating = rated_rating_groups[player.team][player] + + :param rating_groups: a list of tuples or dictionaries containing + :class:`Rating` objects. + :param ranks: a ranking table. By default, it is same as the order of + the ``rating_groups``. + :param weights: weights of each players for "partial play". + :param min_delta: each loop checks a delta of changes and the loop + will stop if the delta is less then this argument. + :returns: recalculated ratings same structure as ``rating_groups``. + :raises: :exc:`FloatingPointError` occurs when winners have too lower + rating than losers. higher floating-point precision couls + solve this error. set the backend to "mpmath". + + .. versionadded:: 0.2 + + """ + rating_groups, keys = self.validate_rating_groups(rating_groups) + weights = self.validate_weights(weights, rating_groups, keys) + group_size = len(rating_groups) + if ranks is None: + ranks = range(group_size) + elif len(ranks) != group_size: + raise ValueError('Wrong ranks') + # sort rating groups by rank + by_rank = lambda x: x[1][1] + sorting = sorted(enumerate(zip(rating_groups, ranks, weights)), + key=by_rank) + sorted_rating_groups, sorted_ranks, sorted_weights = [], [], [] + for x, (g, r, w) in sorting: + sorted_rating_groups.append(g) + sorted_ranks.append(r) + # make weights to be greater than 0 + sorted_weights.append(max(min_delta, w_) for w_ in w) + # build factor graph + args = (sorted_rating_groups, sorted_ranks, sorted_weights) + builders = self.factor_graph_builders(*args) + args = builders + (min_delta,) + layers = self.run_schedule(*args) + # make result + rating_layer, team_sizes = layers[0], _team_sizes(sorted_rating_groups) + transformed_groups = [] + for start, end in zip([0] + team_sizes[:-1], team_sizes): + group = [] + for f in rating_layer[start:end]: + group.append(Rating(float(f.var.mu), float(f.var.sigma))) + transformed_groups.append(tuple(group)) + by_hint = lambda x: x[0] + unsorting = sorted(zip((x for x, __ in sorting), transformed_groups), + key=by_hint) + if keys is None: + return [g for x, g in unsorting] + # restore the structure with input dictionary keys + return [dict(zip(keys[x], g)) for x, g in unsorting] + + def quality(self, rating_groups, weights=None): + """Calculates the match quality of the given rating groups. A result + is the draw probability in the association:: + + env = TrueSkill() + if env.quality([team1, team2, team3]) < 0.50: + print('This match seems to be not so fair') + + :param rating_groups: a list of tuples or dictionaries containing + :class:`Rating` objects. + :param weights: weights of each players for "partial play". + + .. versionadded:: 0.2 + + """ + rating_groups, keys = self.validate_rating_groups(rating_groups) + weights = self.validate_weights(weights, rating_groups, keys) + flatten_ratings = sum(map(tuple, rating_groups), ()) + flatten_weights = sum(map(tuple, weights), ()) + length = len(flatten_ratings) + # a vector of all of the skill means + mean_matrix = Matrix([[r.mu] for r in flatten_ratings]) + # a matrix whose diagonal values are the variances (sigma ** 2) of each + # of the players. + def variance_matrix(height, width): + variances = (r.sigma ** 2 for r in flatten_ratings) + for x, variance in enumerate(variances): + yield (x, x), variance + variance_matrix = Matrix(variance_matrix, length, length) + # the player-team assignment and comparison matrix + def rotated_a_matrix(set_height, set_width): + t = 0 + for r, (cur, _next) in enumerate(zip(rating_groups[:-1], + rating_groups[1:])): + for x in range(t, t + len(cur)): + yield (r, x), flatten_weights[x] + t += 1 + x += 1 + for x in range(x, x + len(_next)): + yield (r, x), -flatten_weights[x] + set_height(r + 1) + set_width(x + 1) + rotated_a_matrix = Matrix(rotated_a_matrix) + a_matrix = rotated_a_matrix.transpose() + # match quality further derivation + _ata = (self.beta ** 2) * rotated_a_matrix * a_matrix + _atsa = rotated_a_matrix * variance_matrix * a_matrix + start = mean_matrix.transpose() * a_matrix + middle = _ata + _atsa + end = rotated_a_matrix * mean_matrix + # make result + e_arg = (-0.5 * start * middle.inverse() * end).determinant() + s_arg = _ata.determinant() / middle.determinant() + return math.exp(e_arg) * math.sqrt(s_arg) + + def expose(self, rating): + """Returns the value of the rating exposure. It starts from 0 and + converges to the mean. Use this as a sort key in a leaderboard:: + + leaderboard = sorted(ratings, key=env.expose, reverse=True) + + .. versionadded:: 0.4 + + """ + k = self.mu / self.sigma + return rating.mu - k * rating.sigma + + def make_as_global(self): + """Registers the environment as the global environment. + + >>> env = TrueSkill(mu=50) + >>> Rating() + trueskill.Rating(mu=25.000, sigma=8.333) + >>> env.make_as_global() #doctest: +ELLIPSIS + trueskill.TrueSkill(mu=50.000, ...) + >>> Rating() + trueskill.Rating(mu=50.000, sigma=8.333) + + But if you need just one environment, :func:`setup` is better to use. + + """ + return setup(env=self) + + def __repr__(self): + c = type(self) + if callable(self.draw_probability): + f = self.draw_probability + draw_probability = '.'.join([f.__module__, f.__name__]) + else: + draw_probability = '%.1f%%' % (self.draw_probability * 100) + if self.backend is None: + backend = '' + elif isinstance(self.backend, tuple): + backend = ', backend=...' + else: + backend = ', backend=%r' % self.backend + args = ('.'.join([c.__module__, c.__name__]), self.mu, self.sigma, + self.beta, self.tau, draw_probability, backend) + return ('%s(mu=%.3f, sigma=%.3f, beta=%.3f, tau=%.3f, ' + 'draw_probability=%s%s)' % args) + + +def rate_1vs1(rating1, rating2, drawn=False, min_delta=DELTA, env=None): + """A shortcut to rate just 2 players in a head-to-head match:: + + alice, bob = Rating(25), Rating(30) + alice, bob = rate_1vs1(alice, bob) + alice, bob = rate_1vs1(alice, bob, drawn=True) + + :param rating1: the winner's rating if they didn't draw. + :param rating2: the loser's rating if they didn't draw. + :param drawn: if the players drew, set this to ``True``. Defaults to + ``False``. + :param min_delta: will be passed to :meth:`rate`. + :param env: the :class:`TrueSkill` object. Defaults to the global + environment. + :returns: a tuple containing recalculated 2 ratings. + + .. versionadded:: 0.2 + + """ + if env is None: + env = global_env() + ranks = [0, 0 if drawn else 1] + teams = env.rate([(rating1,), (rating2,)], ranks, min_delta=min_delta) + return teams[0][0], teams[1][0] + + +def quality_1vs1(rating1, rating2, env=None): + """A shortcut to calculate the match quality between just 2 players in + a head-to-head match:: + + if quality_1vs1(alice, bob) < 0.50: + print('This match seems to be not so fair') + + :param rating1: the rating. + :param rating2: the another rating. + :param env: the :class:`TrueSkill` object. Defaults to the global + environment. + + .. versionadded:: 0.2 + + """ + if env is None: + env = global_env() + return env.quality([(rating1,), (rating2,)]) + + +def global_env(): + """Gets the :class:`TrueSkill` object which is the global environment.""" + try: + global_env.__trueskill__ + except AttributeError: + # setup the default environment + setup() + return global_env.__trueskill__ + + +def setup(mu=MU, sigma=SIGMA, beta=BETA, tau=TAU, + draw_probability=DRAW_PROBABILITY, backend=None, env=None): + """Setups the global environment. + + :param env: the specific :class:`TrueSkill` object to be the global + environment. It is optional. + + >>> Rating() + trueskill.Rating(mu=25.000, sigma=8.333) + >>> setup(mu=50) #doctest: +ELLIPSIS + trueskill.TrueSkill(mu=50.000, ...) + >>> Rating() + trueskill.Rating(mu=50.000, sigma=8.333) + + """ + if env is None: + env = TrueSkill(mu, sigma, beta, tau, draw_probability, backend) + global_env.__trueskill__ = env + return env + + +def rate(rating_groups, ranks=None, weights=None, min_delta=DELTA): + """A proxy function for :meth:`TrueSkill.rate` of the global environment. + + .. versionadded:: 0.2 + + """ + return global_env().rate(rating_groups, ranks, weights, min_delta) + + +def quality(rating_groups, weights=None): + """A proxy function for :meth:`TrueSkill.quality` of the global + environment. + + .. versionadded:: 0.2 + + """ + return global_env().quality(rating_groups, weights) + + +def expose(rating): + """A proxy function for :meth:`TrueSkill.expose` of the global environment. + + .. versionadded:: 0.4 + + """ + return global_env().expose(rating) + + +# Append deprecated methods into :class:`TrueSkill` and :class:`Rating` +from . import deprecated # noqa +from .deprecated import ( # noqa + dynamic_draw_probability, match_quality, transform_ratings) +deprecated.ensure_backward_compatibility(TrueSkill, Rating) diff --git a/data analysis/analysis/trueskill/__pycache__/__about__.cpython-37.pyc b/data analysis/analysis/trueskill/__pycache__/__about__.cpython-37.pyc new file mode 100644 index 0000000000000000000000000000000000000000..711958bc8bae223e1f685441a3322d3196351fc2 GIT binary patch literal 414 zcmZWlyH3ME5IiSw5;=*Wp`*AqMQjRaB7~xdA_bxe)k)~&_*V9bv(LImq`2TW_y)d& ze@M$;P_YgoL1I=rn%QEsn&o5?5yZ#p?Rgjw^4XeSpeGAlOOXU&Lo@cP zpc$c9;HntSMd4gMPm)u#HhI!qJ{3T>Qh=s;S^_nxlT}Xb&N?U`r-S`~bD=ZN_an}? zV64>Dk5Mv%$^x1QpY*QK2KD(-Lzzm27Yy*G6=#?PHZ!tzhmdz-hS!gz*UT~UkwNik7 literal 0 HcmV?d00001 diff --git a/data analysis/analysis/trueskill/__pycache__/__init__.cpython-37.pyc b/data analysis/analysis/trueskill/__pycache__/__init__.cpython-37.pyc new file mode 100644 index 0000000000000000000000000000000000000000..96fe00953bb09751c9884be5fc6df10bb4eb2af2 GIT binary patch literal 25214 zcmd6PTZ|mnnO;?Obx(Is&rEYTyi-z%5~&%991e$)Y|V-^DUp(COSDYV;<_cer>CoC zriXoFRSh|8_t=g{4yk1A_3mcZu;c8KK#*)MK;qqCfdC8dLjoju$YX$@fB->)1QC*b z*&r`YfX(;)r*7Rdq$mSUwrA?psZ*!U7D(GcB`XO5AK_D_Nvvo4M9VWkk|)&3vm+DYS}} zVr#TAinI|s-z>GpDq|8aG{;*Ll?jO#o8{K7%C6RAWwN!qvb!}^nQHB+?2&h)&AqLC zm3^)KmHl`w*<<#&Jz7Wc!rAHn@8`?<=~_Vblz?9bUp?Smib70dp-eat?D*t7Nv_VhdmPU%*$eh*djhfJ_RIDpzB~bl zz0#a*ovfS`NW9uS*_x}&wN6z|wdO1Hy5^jGaN0TMJpYg*VqVo|ey#V@qGegWyY6^5 z8_gykz7_q7e7d&kShpLt)3H`+Eyr?eexto&d3QbEY0XNGg?guZ*KMq<`U}={ea4!f zJ2ig-!K>ENUF(a^dV9rNU$WkD92U3GY}B2$=PX!nT)lAI@*Kx{=hB-OFJHZAEq7e2 z)o~rmcKlkS>7n#bh+8vyFi~6bI?Z+8sWw{Oj_V^STVJg;+LsY4Goq#+tbE-?V`c0>8&t!x0JB^!8vw>v$ zty)oKZ z((EkNnpLNLJ22k;T9CbZ>FsyV2j&|WubmGv*Uo<}C||g8{(AMjD_?oz{2Q0vxpeL8 z!Jc}pS+Ck|?M}7p;_Ib`R9Q}vTQzs3(GGU`fSHFOXjRn^d%;+%=GRxNQ8V}3ciXjA zV|!^U93)M9?c66B+&(cF*a%jB@?-w~`4?7Bym|P)|EE_Eexc3`{P8}5>rGtV9D=@K z>o@c6xUX$#n}%)ljR$&P^Nk0YzWz|}8*Ap2c1U}uPib~$zXrrLYh{e=o9Nwo6X5f# zn#Eq8NcGZMb~`Q02mY=!ZaZyjaZ%u)I=r~3kk?x1*4$c4KCY@)3-XPWrV?g}xKT#0 z(F5|b)b@I7$#JdDvej+Y?gF{2M%(i5bgX)(g+e(^e5cj|9;ZIX;Hj?{>dl(xEiA_9 zv1oOc)||ROYgreZ<=T4F_bk7o+Ajky#jvR8hq4-Or`>Yee%P7I?lAz<%_Eq}x(9I! zGIe`7Ffo*?f$80H{lHwp^ukOTO+f>%(5)bY690kbp2T;*cYnU2x+fZK&#yI` zCmQwD_0>+*IT=%nRIFNUv>Sf4Itx-@;rq|y@^T2YLSE1510!!D^${bl#@#mdVVpVM z5e7^FLx=-1@&|Vuc{Ns#81xic>^SnN_o?Aer49_@UvV7Z@Q0I9I~U_Yx4Z1l;Enqt zgO?Bh3Ce;5seZaA_;!{7(Le>9q3*K?-olk0RT5YJIG;J#GO8~lV(TmVroM5cuU*h? z9NaYe`UB0!Jl@j#5DE@x`!Ke;cB^zY#ti63ZfuGZ~>b_bcJ(?(au6y2(>xSjQ`7Zf@7sw~F~vXVB`A+Urb%QckiiD+Zw z^rC(cDC@q4w0hJ-QZ-)E!^4R5wKeqers0-tV^eRNc4kvY5A`kf#JpU|wYr>iy{E1^ zzI8-B%(l9-cGo+?dA$mT>A1bqZ`RuFj&G453nc^fa9xmr=3R0}-wHVtKM6}Yg1(Q0 z$-(4X&5m%D_d1QXf6;Y2?u;Sh2Uh5<4EK3Ybvevm?ec@F!!K^2KOV70L*H|Tsk>(o zuQMU~anADReF=f1RdidcfM!=R_QXAH8N!`yepINKcGk{85Xn+>!#!sg>>}E^pVk@F)wbrd#+qS&MN~>nwt~J*kNEXdz=MKmzd4bSUD@8k5iQ`kO2}$*(W$SKd z-LgCF7kukZt?d)|>zbmmMZ#z6u0vj~FmVx|3C&ii^0j&sQF;-dBF55M zSCpsa)X*)`_>eV8Pt-dQ;r_8=dz4a{HDbDGQUIV8L1 z5m2zJR)tAct0^+X33v;aH;I7w%D-_vAO7_wuE=zljY^JZFlTZ+gE`d!1Z_>tS9HJ# zGMEoi|BUWl!2993Gr}Ul;`(^!u_R6Jji+kEt0%9_YE$j6lH?Fg2Dz=M`FDSSg)Lz2UI#2wNfo z@>Q$fOVyq9R=ta>X0@>paXhSg3k=8BAvWr3z~i;diUGkAhbkJiORv%M_p1nkQdJ~xRbc?!vpaPZ zMrgZlGq}Xy%M9LS@HGU%SU6+qPO}-!8nSU<^hIWxW5Ai`;PS>1;KAO*{zdBOl zA6t74m-mMV(j?z8ovf2{Mx4BDEa!yu>y?82tPp>_QVfazXh{4^A@LsziN9#$U;-2N z5$fd7U(3|Xk+RE~1S^4lj^yrCS7 z!tl{W=#qCuA@(G%MHcQ+KeipAst9I9FJNPnU15sJ2wj_lrUho!lU}vMoh99?H zojXeTN7OJ#MbIw5mRzNWvxU|&WNvR2HGn-*kRywcOu|0ARp|?bh*0%O$$5+9PS)J) zt8)trs%kJCFh%Q3(z@wk;qVOe&m=8PYwK(&`D@%GrQ9!)HNOi%$a=$BZGbPw&tokqJ2xkZ(g)QT9v zmugxl2uVX$Yt3cV4(akxQ~X*ehlJ|BwJg?2eM~9Xh8zO=vFu}|~z1zUB zV>=cX!`dxK0~TyT&|y;$8AiQCE3;BqYyoh09N;|FZl!En)+I{QZM@pnkO2tQFc8Ai zFcmQCmSNMOKoxh?Uw5UOajwS~2V0j<@ls3;Y>vV)YnshsA8A|RxXnDi9B`5KkY-`F zut*|@8b81nvT4J&IF64mCT{g{)hyDcRgeh`8t8=(ixeDpK%g|NzJg}BcN!jO5*sx9 zHPEoxL^DDB>aY~~Zh?T-V7m>t2rxOlnKc4CXd>Lv)cuNuP*BMKS z7_tStiMKz)rG#*(%*kI-pEP=tF_ll`@C@?v$2%{U;T=Oj3jl3@5Y(WAgqFa=c!EOW zLzIKnfQ;Sf{rM$D#;MbTv)-{H8=8_)BLmuT3%Vg^32_~aaX>GOhf$~xc_1-*KpPC1 z5YgrA*|Xc2B80}dNMbydOy)4_pb9vhyiYJrwU)x?-0a+3QWhUV=E}M+&ty{LN}&x2 zqL-XQTn*LWumocR`sdG)P1!WW1V=MmZ(lMn19@ZhF^iAl(RmRMg#UhR6X^Ir>t`P7 zn`Ym1pT`r>7y`C=2uRq^j0!6DZd?N}+nx3aArYa$ofG%S`$u95i^<8y*y2H6a+@Y} zA-|pFc!+AkuM^q?!v+!9tssC|i?9exyRp3NQU?h1jUa0~?M`c02+!WGf>Hik;h%}vZ+0K%lEfqV9$4p3n( zKQI`VJKiB}q@Q!&>mxj*wK?;N@|yb%+h`khMwvBx&k}f#1)P|_+|y{1GVx-$0}CT5 zpyU;Te8?OElRVy638Fx+1^SZv4ZNvjVY!A*HiSI6+p}RjiK&Wl$--R&= z)7|Xdd?YemA4DcRJ_#~ym*x`GQjq1y!xG?DaTgQVOl}CEVk!&t+kt*(7=GD1#IpZ@ zPrVWX4Omvf-+tX3z%|Jg!;du*hC&Rxhd?il6aKKZ`WH2Y10_-qLOJX2$)JpR=>_%@o!kMjmQ@9z5wANCn%)cLhw@ zkpr4v1bhDXeGr2c@G4y9##%oE8~3%%>k9evj7g2PJ=utXPkBJ0QvrJxCs7+@y?R~5t z+Tr5Y;-Zo_L-RLe+DMS3&#RQXh?v;UTgq@Qsw;%E=&=Q@Rn%L^ zcTTJ8-~AN!m~gy^BSHs~W`<_bu!Ykf+(@!sb8B^nl9)B!>YzSo3{VI-tD6><$^4mx zxSeE;v$M&}1rXMqi?m`dC~doim9qxK!bkvD-HN~ev{p@r?0M$+%wPEi9=E1{1&tJe zT%ZP9ItC-gnL)sMzYIvx;vM0O5w)I!g$paW@FIBW;ydqLdhhC`t8o<{OjCa1D*9~r zJ-)yYp*+5jMS{UMH)5-XCFHAj=rV0FJ7Y(0`XU#Zo}kmhczaKuPbNq>voJlx07Tni z?-|ud>Y=m@;azP29fjRoo|$l80p$vc2@Zpy!%-jun~h8A=6;3m3)fJlpg3grPFDU(#d@qdA)cwAZ3_LU%j!D4TWgUAfgVv-FlVeo|4Vw*pN+b#&N!i%iE7&1#1#F$}nVy{A^15E#np) zb#$0;*)`mVGgz)LX6FKZc!BR1#n(Cd^guX!3@aY8PLb1kvndHKBFA?e`s{eKuQxj& zwBA`nqE90a0?{1Xs4pluA~b^A2&M&@6(_>6U?eQbC9=7cVk)ID39=^;l}_8R0(B#KhgiJqts0<|ECA7{$Tg(m&3^5o<8TfazPM2-ioR zybZ&+e1)~p{rF109PzU(^JA=k-pk(0!ML5b3m@e-bN7t!xoD5#dE}msl>EKIvPlOv zq~!07+#^E4c#fySz05uHUQW`obZtXg5k`cOe!gGm7ndPS&rH_-80e)I)Y~1{rNzKO zZ8rv5WZPSDYu(ipU;m=xI#?Bj5iPVuE8EM?-@#x81KcU=wr9;VIHe5MDYG(JHhv)~ z5nq`F%lb@!Bw}{KJ54PsZIFMI%)MktqEx8 zU?%Z37-|9_!2jvkx9n5uX3e-BUpH&sp=v6Daqn5HTu)r}_EL9%{kEE!$EUa44mAUt zQeJc^!-vp?0TRinDUxG9n|NMPLCc=6>FQ9Kt>J;_=zJA|WGqy7CYsx(+vT zyqM}a)x&|FQ)?gUy`jANVEq#ike`W-7N|~Fgi2nh3ZMW06r#B2evA1oFi5&wRt-(a z(6xaMzxEkBoRkxHmt7^dmF_Mfvhg9htHi4JRA>7dv~<&*z7LzC4cj8bCIkLc(8rmA zr_6mAj{2GJ!B1?{yicRKif8Fx1(QZL1W@t1Cg_p5oS1I_lsoM67y?Z_RS)fjA*T)l(kNN>hWkSP*xzJoG@@qw~D_j9Z*f$$y$;Utsb z)pdfojGP)In7lr2DEkI19A`fx$cGA!;m!$0^bojH)Y_N_+(quVaV$nM9gu~DY3utm zLYI=wN;KIbXqqV}Qn57-fYflmi^2m1QQ;{>JR<~4l;F949l0I>mV^`42o_=t!2(JD zmjXsoTnvmz4-jDdOBBEHa{vPdAkdmWihv<>Mt3jYhm9-&3Emo;86Q@(EeL`T3PF$H zJz_F<-~0hAnC%?nn^_u?{M;Zw#jHdhscav)wlbR|KwqFVLprq{(BLrm`}jVOs{p+N zqFj3%@glBK_3W2`p9PS3S{C1NlH%abOx9AhbX_F0~OZYvJb?hg>yywSc^S5p@Be`5GuTaDFC%L|6yjc zuvy$3-PjK*od1D_sZ`o5^-B+ohh)`nK)D?uZ3aEjF^R@D$1#1z@a@FLzxLr5DwFR| zH_QDAJBLZ|lm3{kAHYHhzwVDqJo`}F+|~XKaDrVM=b&#Pr_!8m%;CGSemTmIh0}Cn#4fMx;X$`jAxEE#gqVyo!FZ1gk>bKnM3@d2s-hL6^ zjSD={BYFGV*FiR*$K`wWRd6Z~i(#HuTY(;TrvYCQ=w{-Cq*OaBu~D-NeKkakwbt8y zqsio0Ux4=pYr?eL$bqip#qb*b`RP-Oys}H5BDMvfqt93mgX$o)Z&YiZHdt8=V zF|Y{;`F@B7>fEp;AmFxXQ$kXc6ekVOqjrE2Dlw zT0^l)y>;4ZX$&i{KpsUEb@zAK7a??|Gz`8;P<;{av>g3*O}&&gbptWl{YnsyU=c(t zZ^jLle}{09ZwmSQQBM=bkL8UW`PqPYd42?Cq_ zK33wmEVP1(6?fk(ZoJVS!DL@R&7&ZyC6da19v;;tAu0vL#w1pJkiVzHA2Mf;e3b8x zb6zq1)^|XNl$j^9IRQ6`LVp5sNx5IXkKFq-Nh?a)uKq3?R!J;I<+pUyKe@RZHqEJx zt`GP86}`V3bGAT1aG%!SEt=Rhv}(`hUgX*bVm8^|)8EVYXz#Jj{r!Cp@;;ZV_OFev zO+Y?UvGNwoXdy|PL))g1Ygd1tYAp!dUWn;MY!A>j4**8o3NQj&EMNc_O>G(d0|KXm z_d!8!=6|TW|4>X`TK^z(TvbeIoJO!HWoDwb7lV1lf$$S;|5AFM$1R5>gp@^O0~U$g zkSS`|%W=lCo})W|;wlnt2Z#wL%oxvEp&Nf#Q~I5I3T~4qcF6hT<@wq9`S}5xjyu0& z&d_?9iNa?I@ik57n70$!uS)6xT!=pc8jOOLh@> z^v#~Iwu!7+WI{{^vspWOy z7`NQ0C%ak_5Dp>da~AgJ*kMMPY@;oVSm&)R)x4y&sGv)`rEmLLgYLs*Oh4XF~ zMO|#U4J@o)Y<6B4K(H=R@iAY&TXSz(Z)3aeO}H%%eR}%wpQbnc4u1K|VD&ePJ&lCZQNsUw#3vr85ulIu2i;R_Bd*go2A8clWWsmDrtUmR zJ_z%W9p4BaQj3|P$FKrMIAs@r)X~&=#GDBiaRI;>--)B!e>?lsg|5=f9yb8m9xEOo zwU{Yy+>KoBj)$(QvMwrf3Qn9ZEUTEGk7xyF?Kaww2&rM+p+RQB3L~8FGWiwg)Jn$@ z?7&J4=AC@&IXx3tMop%Ns7o3-o@7NWUEsTD7=5K%yQ8C-`WCp)FVo&PIJ< z@vU&rU$RFKzLNF&I<_p!;tk!%xqn$U75JSFdU?lj58M4nPxV(qEr@A=E}j zOI)deUMFN)^Sl1cPbFZp-V0aXtd=qwKxS& zfzYwAX2aPS+X3cg=M&Cyxi@+po1#|)FTGe@fyf65EYvz^MW)FI7BF1Uv0!4QbQp=P zViFMrBi1s4{OiqHYss#i{Vyo$eAc=kTGrq8$i|PG%i`#RExbd z)X^nLtUX3#@z&iRAzP^WRE#?~w!r-{QWCY}7lqeXoHh;#aL@iZC2CbX{JEkN2S{hq z&5Cj>b17R9H*EkugoT2Sf(>6j1~%^R$*%j_BYU7 z@e@#X$KPQyewV>d7|?W9nT!vsfP|^US$B(XKVl$`27jCR*f3a1{7PZH9X3tjfT18N zU7&jE5^sLSK|hCxcAVZP)J&Cc3Y`ylP2ss{9ERR0f3V4!`Vm}{`Z2ttbf7%sz(IS+r2&K9)P6g~Iin&Wf-iLkkr*1$idw$g>Vt z0_aC@Q^+X?h5KT8K}uH4crdxaDImu^1u)B*utn$SQ~(zO(I+=Y@CL>y8v5w&2os;S zmbdfp86BB|ehFV)SfKm4Fkb9unM=|~>0$sqvxN6qdxtu-MJ zQlk#0aXu91V{QdrOD3{ylL|yLrsL3!PF?w$4~pK~)WWz#i_DzLQrMvB4=9^K%Hs`! z9&e{_%vi636TLDwEc1175{N(j!ZmVz)h`@+(?TOV)EQ*sU+9x-PMCZbniHHX-`$S69g_g;MR z7RG}?uEio`Fn*4R4GCi7r@~Hk8_O58P~GPd6tbvE){24YabR#^kH*H!wA=MTKk34? zPx%}?jeQ!Kes0+SOU@n8)WE^wm-!r)qiGy}7Sz-kYzurIi^=SD0 zt?$Cm?|FQS9U?Y%JZ$M$+@K{ejEiiC1y)czEwzusOtEM7RJr(#REm&@t7u$S{tx^#7&7-ZvkD)TsXj7thDx*00%WyrdTExbuUt{dn8kwGEyzZj<2gdJ~tdqHq%&x4wa2LP|!Qq3qZIWzc)SxT2Iq zVP)`_#Nj5Ptt;e8~yz2hw44B4kT&J6ZNu3UMJP^%Qr z+lgDeoUlY-_c=gG5yfyP9NkC*{pK+Br_bmD2gyW4D-3QVQLTvSGN)oZ0$hv%X$OV* zzEI-YAK>*BX9Z?S>c^BH(2?lSEoCR%=}4KBy84JMgs)($U!)5&%M~$heae2p`B%~# z20uT0`t<3+QV8W1j!>%BysA3rW}4RR$2ivevG;t8q}-YKO1L$VUiP28ozd5_$MQ_r z&bXL2xI0rFu_MPLtZ|b8w$Y|Rd)!)}TfxF2sNQ;akxQ1?28}@=y28$y0x+1+EQSrY z1L`JpLdai21IGuU@)nT;m!^&P_QxF`0$q$EB*|n#GD&|A5uyl0XK`4Avzm@$ba3hk zju@e{IL;}`W3mCUrqah4t4*_kPAuk4~+%gK>CU$V_jvwfJy>8$1aJa!d*XlId9 zw6T3~Er-1Dh(c*Nng^GZhIDKOUMu1n#Z?0P!r_qcPlgyS2b<(CyO+_rUhe4ZspX?b z-52l>XuUCPh*ifNp22YTO3C93A02f|!AiX(R8!Tf@ad0bg>Z}zU`Gfxh$%!lO`>f|W1FjajEZKdE!P5+$ zVL%LUKVVR0U^5`|b5|Kqv{Op>4-pHp7cRbY?Yz>mDbKp(9WH5;`%f6K52=2|$SMbz z(6>Y^2>EACEO?HWiW7PMM)Sq|d~rPg<iB;wX; zIaU$tm3_m-DJk4@x&jZTzPSc>uw@9?%lZm-tU_inzYjeT@(Ls>UX)x^Otzh6Yd;0d z&vP@G>(?=jF&SxMvSbR+$Ey~?EmpxTPWYV@yl-;}#~y?_P7_$_EP-Eyv~T_uZ08B% zrxWrxkd?K`;h57&a!jC-q|mZHob$M7h3z!s8Bnya&Oj2c%%O0AM{Q8QNUV?yvv$Mg ziC~-r%A7p#F;*q`vDihQ{96{*UI?p*0cg9<9iE>;dnonU4vgX?Q#fhh(+(wATzm_g zIpjm^tp!iNq?IR|a z;+rAxVtaQ@E%Zej%?R=V6Zm{&1P`hHPJ%*c0Zw~@Dsw5s26~r?g=c0Y7*S(-DvHeG ztU3;XGOd!4i0@V)@dIL&PQybnVFj>=G53q^X`ksWdR{3u3x!vK#R zhi6Wm+=f5ic}X9iM+97hT<0LgB;^kyPTM(h7GKB7CplU68=N1 zim@SHO@A~b!241w$6ms$-C31c(va6D&A0ve) z#m3ZiM+y9i+Vg11o5zXJ2~<(Te}))NemTd4K}jqnc-FaIFf_;Gyhf?5&)~@?dA`dEsn5OWfNWFa|c06VpN3)p17bbxy z@rjG?kCB?f_c=Z?=bx~5xX_bWP3L1w*T?aY^y?w_R5h>X0xcz1MiM?kHV)s5fan?lBxYRPCdl zP%efJ=i)XzDtz^C;!fn?fmyWa^}{pRZJ;%2j{6*gBM2zg$Nw*jK+jVgm)GgqgJngI zr8+YnE8R^`TrERmlh@dg&D22wvDsGVw&N+sapJTr&hboAIfxAf6u(A z4_tA;bw~LXfh)UF#NS@_sKHu=gX$avjk+g?q`I_=@sLM&^P}WP>K{L2VHSgD8N9~e z41?VWf}A=)TupO|DlS=*n(&;rg5D`Dz-*ROLi-zoAZ# zPYqA3vtW5#9g}&Mo%XIEfEOpY1bsNCaFz7KMmcLDW*9}oG>VAl<Hd@$VbugT*PRI(m7kTqqXvMV9h^0UEi~4*&oF literal 0 HcmV?d00001 diff --git a/data analysis/analysis/trueskill/__pycache__/backends.cpython-37.pyc b/data analysis/analysis/trueskill/__pycache__/backends.cpython-37.pyc new file mode 100644 index 0000000000000000000000000000000000000000..1fb6a2f2e8eb2213fefc0e57792ae44e86ddc48f GIT binary patch literal 4073 zcma)9U5p!76`ngY_IkZuXOnD_O*d(Wq$F!8>)mYr8Y_*L{6He2q>!j33zgUFJD#2F zc*eOi-o}=VP&TMrK_y-)B>IvdQUO9j;-NwUkw{hP122FOE%Q*2K$Vgg+KL~6M8bFO zjJ-}mDrWcGxpU{vx%Zy)edin>8y-$8xZXMS?HAw7D$1Yfp?f9JnT7aQ(Qsu>;f$*e ztEqD;qrT>7a~kftqt_C135+Fpk`M3{AG8v4NmHvbp5{Z>*xZ2ZXZSGsDW2u)t|@ba zd=p>K2QZT6oB0NwLT|`SnS(2AF2gtSB<2kBQJ%m(%eC`LZcFQ}v|$*5Xqf(z?Kt^` za%IVMxlf~CedJ2h@N>djwz=sWwQ{g%qOmJw$M6GmeDr-I)?ri@J5`xS_m^$>jyd zEOn0a`Py7vdcv*RwRaNrbwNyuKtFWWi`ts*Y*A(Rd zc7at4+g&z=ZyMExTM2B>HN2`}ifV;UOYbdv*>t0k-juS->Fm<3I3jO2kui)4qkZ5? zLQ;^85dY_B+5pW`LS#5LfxfoH#2D9QKe42ASeyBKIBP3oN=I2`*A<>bi)TJCuH0aw zkd8{=YO6dws%%mC5PBKNFg=lsD{XZQy^eNeHo&1**a_vu{T+Q(X=^Ky=0sa-CwLZ` zwKl-JQ5llDX>=4Tk1L1p_2uOklr|G@wgJ1gHl)C4>ss3{e1fU+cgO=41lPsef{f68 zn!Q6$ckbQ4e?J~lX07|Ly|(M-^$%`Y*U$as^luyLto8lv_rCL^FZ`w$&-t)ujsNV# z-+zDmuSM(Lsr6f*|LBdPReSx|bMFp+T+HdA?q3!`s5zb`$myYGUa9AjA#2X1Tc+?z zbO~#nYcCjqWunE@MUb0NL6n~OG;&@=8e>5Kth|Tys zr-@PA#3pLSsF9;;-{=;MhM9gXwokS@AY z%$0#rs0r!!XuKT1b#Mh4DgLc!tsZ#w zcVh~ee|NrVJ@wJMMeD=2-rw?vU;eac-Mst5tEb<4=a%)sww=rWdJkdZYlo9>{3`eH zE$bII&g=j9^*@W&f8PK4-VNvO-m<=TbNR;;e|) zbnhNQ5+pi6BC+)nKM#rGocaXov4jE!QB9uZC+Nrwn-UuW zC18PCrR;_ME~(-x%?wAjbF>b7L229;r`zsj7OyihQv^ZTM#XTxNq zUJ|9dA7K?tlNY5i z%7c<36c0GFSVy^1;3`VPnGS1a5UWrqN+@pzF&Kz36G8c^RVFqR;@C^qMjEd%7-QKmHkAmSdjzjNa1NPn+WnoIc&(9Mk=jV;dNmSy2DO~(Q z!Eb^%y--x#)AKZXk>6s(U-TMArR)ObW@X8+tCAbBCCPPbJF?w~Hw__>wG7i;wuR?X z1~Xo=gGIR&iJxxpDik8{p~#2cF21qFHKkov`4hQ1Vym}4d9e6#lNdcCGj+d!eqH#$B#wSz^PcfFLu!f z4(qDO5bI(d1!v5$1*5(%LqANiIwsiGFz<9l|u9DlMI5)RdahvRYZ`nGi(dz$8^BrAK3AP=hHHi=y%g|P*?4q6BZxLnSxM4J@l z;b_+@$bIPgp(s$Ged=rcn7`0J(Eq^KKIJduA?SC85_PqaprtV6=*)2D%y+(X;b)CT z&A|2VFaG)U4;zN@ANn)B9CUt-mi`MjZgdQ8aw{@pt7Dnewhh20za)VC2H&%`JWr|F-xgei`Ez`4zq1RlW`BCH^yh4bscv3V(xNe_@=N zovYm5H<~v`-_{soN)AQ(B#fdK7Xv8*UkOg~KdrfHG;qI|#*T!qhEk+0J+&*E4r|b(c?03`QA$uqUtJCuvrV!@kEwr2KCUc>y!O z|K%+e4{i;@!A{sul^;brVbB})5>MPJOgq<`8My^x(O%ckQVTc33<#xNxZnr*e?Gg#; z`YIJ{-Mft76h5DMbu+C&7jG6=gfOKvIPd1n3@nC57n5ti28 z==!6cV7^i^JcJ7|Kax1#DHG4aRHdX+&rfMM+xT=k>4B7q%ql575hBd7lf%cEqi2pOEDQ}_|Xqurz&qDc!_(RHr1m>9y5>Ve#R?)YWUGyF0 z6#bGaP5LG;t1|VufumOTWEE>_c55C=m;Ms)9al#PxNNhcRm#pkLIB*S0O(EP!5cHl zYGk-N8(fSL7iR;o>?FionF;5qlWpc(d4)1or&>(SmDe1Y5upL)KS4{05r$bWe(MRb zjY;#4H8#Gnb*!<73Zt_rF9nBNf!Um?gazOO@q)F3$WPPuL9XQigUv;tUg=GDroQyM zVABzPU6 zN~5Q1p(F}01$XO)$D+^hm!iT?N08Q80q*)Da5QIzR}sU>$LLu5Xeq<33%~k$`?(4K zB2TmA_Sif!`_|a}hxOFi&tp&1%4%V%Q%OG%nu-uvt=pL?mvH^Qb~i%gpp17(lR^A0M)MrANj;jpq>B7GWWPmA$?}G4$yQWhj~AonRg)qt~Kq7O)HaQpxT+4b!o@9u-hMLM{P_b zJ#9>q{w5$s(?w6pE;{{PM`zxeC?il;jqR~BHqJ^q;+?LwA`y%|9;OOb>>(NMsAQ+- zBO2Ja#=^uisG^OdLEAuM=56!HObkx2mhUs;j>GzTfw# zn%9?>DhATOZv5$kmya37f7r;Ai^>&b`41GK(K13aw0dUWYFQ@hcF%6vcsf0&@3vgL zb3?D^^-HZ%-*5S7mqI@*hn29}bz9{(jj$FjJv73lsIq6bs%S5VCA3RXH7e~nts34O z3hQ`NkE$54g!bXkN868<&|XG+B`l*|R_#M*9|_G(x~yQbSV{b4xnMUBpIkhJ50Yz&5BBpaJU z(MYzUhMudTCTk2g8@I5^O|H^t_qsz7CtLkBHP@xiaP&aL-L2$O<6NiNxcK5r7tf=3 zrE&W~<3=Pbxh8N_2W&FoP7wjB1xNfgBW(NH8CcxUg%qt8tCTbci6dn{$#9%Obr*v$N&w?*8( z-HWp7J0c#6Hxys8`djgxs26VyhvD>1`R2GktuAAOgHAh%bi2cq^TbuIAj=wx$Y_~> zj1`&?ft5KMVB1F1&io)4wEIyIWR)P$yhXhh1b4^np6>aJA?lbx6<@4hNcy7-qj+>4 z%um|A-ubw*HQpKq(S<^ybG6NxY}deVQgV_cStxun3|=`hQ0{$UXX;$?zis?q9t?s?PLoay(^M@8}1P5E?WL54%9vm>_+Xd ziEpKk!Z7QmZ#w4ga7H{t&zFxdNY|TmCRcsBLnr+hga+?!h=Wdk{mFQf+Z5)Yj z@NHp5G~jAeI@JiWdrnq$bNA@M{ityT2&o;+N?7!nfS?kO#=j|6@DAK3=rvIXmahgu zH%j6p>IcCk3{p19Fn3prb>{|jY>Z&ylUXghj0(`97G5u`!*pq_Z+hu778=&)5*u9eN%I3Mfo$RD34_BA*i@x$(6+V4H&$Q0OKTe!qz&Wwgo<($9VA)ifs8c zDI-CBxGuy{05uDCE53uFH<7h#w_unnQ;Hyff+Uht$oNGxW@-#lY^W~H3Z6=J&rDU< zeo}P>ot*d3kP#z9vNCiH=5;znn+VF zUPh7mBQb=@NFFGzs7dD*QcEhGBLm*xYP93N1&xEQ)m*{%9poaBWlWOlU8tO=0zBBZ zhh-=pLPIE*qXIn_@+=9407{&l{ChsBX3UZ*mLSqo))OIB7onA@6fCe)*wnzu<33j# zJ~f{*I;ZeyRtg^=P>wrIM+1Tg5bO~&47qWc?UrBu5_!to6^4JSa}ocOsz)-5fQ!B= zQDMinwT-~!2tA*}=Wzre6K(XZZ7)TLg;31cE^Yhp<5p;W%8<+6_>ABKS!eku3mQA| zH5O_&u8h0=wjy1blFaIlwfM;MsbLW$LQOCyPbEqUKSG{v?iq90+^x@Vj!KWksqTn} z6e15;a)lp8H?z5OCaNXK{tvkXrRZDBZU@gIkylJoQ=(5jPdz~>yW9eZa55-?LuFCM z_xrGt<3&=@=CP7mUBt3L3r)1mSRil|X51Vzp2iHPRQ49ZY}6ho-etZr&x1}hg-#2{ ziuNGD+ZHT0;wDxFgM*v_`%%h7GJVE_< z)btl>oLX0@9nH3QitZ7{ceoIh5YZ8%BCmsbk;oM$CC%JKFVw$Eh?ofHb(fb?FxfK= ztO`1U#Tv*+Ze9bsW@^#!eF+`pq*@1UIqqj}5{bS9Y=_b2rp#5agf6Xt>}E|0qI0nZ zeuSQ>1|HKIxKdos%v6uYjwweBpOPE-1W2ElGwv|L;{o~;pdNidc#8VA{n$~KyHJ~4 zhmUX~>Tnkd^Fa;jg3nGGf3WW2gBeJ0;bX16sH`})F^gJgPQR$E_hOlJhW*jyPdOdu z_Ke*#vwaZtYwNusbXQ(RB_9r!y^g_|rw?-CH8eD@rgsepNP7|nPj;muCSSbp6u;$y z)PLTAAXHl3zo>TFB#4QSBmNZk{)!5@#{>Dq>KYUK{_TW31OE^u-+CX%^@;o71)ODh z_*qZwN4Wb+td7KlPLZOOPP@u{tgcytI^R8M^s}$LFsWHy>YJs^S{Buvo0( zI=cRhoXfK@W9@jfP?{wryQ~}y#{Ed(rXlm%qfs;nv%{@O46nsIvCM;CC3r3HH?Roo z3Q?80@gR)uXRcByJ)LPml_Np2nwpG{3D9coVdu|{Q*L1+}xzjo(({J_`W?UP5en2qVVBZDiBIJt5QR6pYSamL33#? zu$*y{TW)!{0E-7ZdJ1w-JBhbZ%j_`TnK{6jInW-Da(jd}4^CoVk-oc9^kv&Y3e0Z8SDZ+hMR57%` zhl>U$VV37m?Dri9BYl_)BRCklaYF4kyq#bZ_wAW?Pd}@nXx|GEE_dTW)18yj&-lqr z7PnARaYS9RsdKPC#j0b3ienTgXJMeK(XVFyy8apGGDj$*>Xe*gNx-+cP_=NS7h>ZCsljhlFapP>+}$^;i$lee_0 z#;L70^{TFFqiSqUqGdGAR<4@k>^}2tgSq*&g8V*b^ygacm|Jj*%SJW#L*{EgpdHL+ zsWo05$8I^17gj^BmfeD|ws>{oeI~|4ZW}Z1(@iCjXB2`c6({nt%rfV+?Mt|oOzedq}`lmqbz8CYpjIV?MwBFJ%8=@=dYr8!(Q67 zKXAK#Bj_&KAG+=woqN6M)m=YuuiNk3{q7Yza9#Vu+wZ=&c=tU8W2-G)Te!GfGr;yw zK}lBO@r1J!w42?~t$D3ZTZW4m%Pn?WORhxA@nv04&kF@g5Dm=s4)m&KGgih4p!i({xK#c^>0^${^6PNJR?r^IR0317S- zO#suIeVgd(HWela(SBRDoTe?j0C!pH5?SXYIEXFQ=>_^xt>JkWPjClCgbS^W$?K7Z z+LEV2Fzy!rjQf)q(bhD1IfN+C2>QlR*4HCF;?t~epk_5KGEnM*fnCkT-b~;%T8{0= zMz`hqp$!)FU01k5fzdmsrY~Qfw}X{-w<+u;*KSDH0X@Jej&EOk`kSx6{<>nsWA`>Y z?t4eXd=yWS)Yy@s|Io4X;*i^g9B5or~DE@)TNmO4gydJhXuXnu8 zRj_^tCcNs^SGp_hn)`YtPa~}nshI0*##XK7`CeG7y@tJlDHM!b_$#QAS~=c3k}}X- zy1GvIrXSTArs?&Ft+MtcN=~iQgfZSb|F#WT>9{i7v>&YiK3lqBS3(?#pe@(&=PKIp zY85WWv6Z@CQXHXf8O1!FfDUAOub6?=>h1%GxLJ|@6dDAEJRb60@N)vv9>2>%(mY!4 z%@+bjzy<4F2hu0oPyy{_1@A*gLYV6A+*7)J#ZZ9924NQTT9={B)A-?+n84H@;z=)2 z&Mu+oTUU3(Jl!F?$3=7q_K2^t6V&2p@e{y&U+e3k7HJW5_+M&tr^S7k)Q4DA;sj_3 zwXxa8<>bgD@zN02alTe-b(^)?0)_*EgYi?mH<l#|gTO^$##c zT&&fI)rI%4R=bInLx8780M3?VKohPr=AA(&#qpZfKAvKEjj+o1V^vG|Q$ckAak*N} zU9Z({5g3Y=yf-ld1kD|OKFM@;z7iYgbH(!@s1k?g9M^>O-hR$;H1`t6K_KXgFL#_< zt9|&q<1qIU#~HuHaShzyd(SzJ=3e4B^S3z;H~9E@$I5IKrACTUvJI%+B?M1uazhtN2CiRcz?Mly>y%5Y+ba@t= zjewi*Wo$NK+BOafWhnt#&DDNDkSA(nlJ|}c1ww(Itf?4_i^_niqB2eQ_MH}x9v7f) zl#hpYsaAtZZVqOkr$oCB1Pzq~S)tfU?adHed4ixPd{pTzQWo+g4Uv_Qr>QtY#Vb_& z9u?=P7|OS5DN98D1+55d6c#shLo>8Ov1FD^%diYXv-IKLGmaMCVZl#O>~pkUdBT?A zg9u}LtZKNXFv)R-la)goZkB52;Z7IeAX%a)N~p)gxG1A8hzT)?x+so_Dbyu#R7|5D z7steL)Mar(%%Gl7XPgwLwkKz?*xQIW!jk(3TA|9sc z{uEQ_c!K|iG=#89TU>q_agP+d(T58a=?(337HPq}IuH!EZS~E_++jjrV38i&QRC!M zTK!y7(jf@@k6ZT#9>C6M|f!5FWt^Qav7V#(_SV9KPTmVIRry`w}Y=nS91WxObtkrG3FAJXk_%zYv;X zF3jUL+7|B@Ln|tTIJv>l*ZL*Q6t}p@t?9v6QRxdFXdyga1 zBOR+Ep&P7U?oVvo2up-1+-*L&8LbKy!|`Y$f=_({=kzDkJM$PTN7i7h`~_d1zYlC; zwTveEz%K0@mHT>w^~X2vh2;nkW8WM7JVqv?@!g4Na*uPiaS3y9YcY4GZ$*W9_Iv~YYBF(w8?KLji15*|!)|97zn*1f>rEHI@v_$hW-#1sNy0#E z(5{j+A1&L6O$o5vIOlXat}kM}f%nazi?~Kko+|o*zFQYZ3BEb@;{uE-uGM;HXhI$P24eV9sv)HLOae(Cem@IR+$%+t@ZDro) zA7JEgA4p-6GFXX*(0b#8BNE-Q?*L-XcWEi9B|1RxlgS|iwS*&Ha1xR0Z|QZVi?m*8 zaFP=A>>#p~^r(X#prKrM1ZFUSysvr4AZ)xrftgaim;1)XEP9FhpMfc4k$I9u1|A)` z1yh-}%lFy(nfnM$2`3oGdIWVUP^IuAVmb*4!Jc0XmX3uw>@OCqPDGzwVHq5OJhx zI_pcA%SYy}5#^PYyMk4QLbMGlXGN6I`RnzaD328^$^psNNk%KzcYxw0pcn)IicQ6O zMq;dbi18oQ{E?i^k{dp9U8HN2PQ4&{(8|pht^f&!*e4aD$Rq!VS|6aOj42XTO?6b& zRQrv*Lfn+l>K3&W=?+lMgywF`pJQ4%pG2|a>M212%aBfOAc@pM$eyyVdM#sLq98p) z(clb>`3MEUzVNrSHMskrnM#v|I$?4Tew@ZW=V0w$0fU;*XBcG#w{-Xy|Mocxb5Gb4 zzO1SIpcN`_4_*ztk1gy1pU3E%JHYx$w!=*qE;rZW`XZ$YGT!@ZHX%WquEW@r0WG8b7UPKxr$iNqDtNXF<4 ze&w&g8UKvuCOj>D2U+lwxEsyGF)l3j=Dz1QH`8QLOL-?S_C6W>It&$=Y0A(l!|M29 z19-*n*KaA$K)EVUk@EXGBS0FXeG{vK^C%+r>8U;s>H1Jtgb*+&kogHxZ0nkGzbHbF zbtHW&TCB}oQ;Y(=KwN@!VXHIrtG*grAn#(MBX5v+Dph8%-i3igj{rLKtg<%-urrok z;$JHH5gtt}Bc-=vOSo=FCAvuq$?u>TaQJY2tjUK}v*SH%hYH}TvFyO8zC|xwrGof- za0|Ie14=m+M_~LCqrpWK#O#!koCM3y!W0?!JB5Bujgpxv^DDeJ_3Sk>yc6`9>iQ3K zUd;H=z@$tb!ieZ#2t0!pSOc0K$_{;Em1JhXG?+2_xq%+fMMjY|w8$)iu^L(f7U3>~ zeaM4-je$kD)wecoqL)~UZ-JSu$UMP>cAUvy2-8PS8etIIRpWcwZf=i7W7A+{TBmHs zPrSFut! zV02>c#C$1XN-m@_K{v6GT>plD&<0s41pkgD0R`eIjV)026wDAQwK80d*`ZsJzLOCZ z`fHf*pxjhSRApf_IU~q!eA_oTB1f2Eh!$*7M$*%)!5h#}cY$yqS>Hq28nv;PNI&cY z;UtxYx03XPIb`t!FN6*GLllR5Fmc}V;mrkZj3oiFAj9ooX3GO^9ZYBWtUOj^ocb9$ z>7~G3DzJ_8WhiQu6o#IFn!KEhP zlrubJibE|Nk~5AFwc3ATeL&iQSYjF?p0L8DA<6~|Gn7$UhR~+0+BOm5fZTxBe~Pydr{gZ@9;=R5k2MpRMea9{_f**+VPD$I}YlG31X z2bH6skbILyRkoNgr*x^L^y#nhBxZn*L}V((e~1ZyA9I;rS0o#fl)hUeevJ8ix+;Bs zI7li4p)A}}f`(!drUD8I9!SckxIIc-1zkqmee%%3t8nc(1*pMPN?J$=eCJgokq|}N z>*S~?7hw)ka0$IM9Rg=^CU|m-G90Bwe}g9p zq4q(h^4J+<1QCVT@6zH~6!>=nZ5y&cE+g{P$cFBM%6rIG_cZ}Zs;qTxn1&)TB6vvo zDu1{^f;59y*>1X-@;~?ZUuo;~-9hge_Qi{K20|)`YKLiTqjo6K4wNJxyv2LQ1tjrD zWiK}BZnJs7pnX+XpwvRW-Rj&x%#&as$9nSvB|n-IACbH49gL-~!`Cxb+nx{KNQwF_ zz;@s?`6G;t9`o8u9@9JfV99NG{^OSNWc&7oznprj5k8ck~c&@PTr%ax`J5S7{$2ySzXJDFk_u3Z-|H2D(k%J5*%$ zRaU9{r&Oqn{sU?e%c*>QJU+;L5Op9liYt{)8nCHSLc}wl;t9^6fMwCj-zpg;!?LU? z>zFlRnO4CnT7{BcnO3mT7l`F9K3U^ygd~=c-=~83CC(30Q%dUN;%)W67|BO{{GUaL zWK-L3#O89d?SyfjSflCUUoZl1BS|fhy;Go)hKqB_Kc}dVwQ`;gAy|jkNLtEAwic3) dA~z_}5zvQ?B$GptAGyL)T1hY9L&W5b{{#Mf%@P0r literal 0 HcmV?d00001 diff --git a/data analysis/analysis/trueskill/backends.py b/data analysis/analysis/trueskill/backends.py new file mode 100644 index 00000000..0fc03754 --- /dev/null +++ b/data analysis/analysis/trueskill/backends.py @@ -0,0 +1,135 @@ +# -*- coding: utf-8 -*- +""" + trueskill.backends + ~~~~~~~~~~~~~~~~~~ + + Provides mathematical statistics backend chooser. + + :copyright: (c) 2012-2016 by Heungsub Lee. + :license: BSD, see LICENSE for more details. + +""" +from __future__ import absolute_import + +import math + +from six.moves import range + + +__all__ = ['available_backends', 'choose_backend', 'cdf', 'pdf', 'ppf'] + + +def _gen_erfcinv(erfc, math=math): + """Generates the inverse function of erfc by the given erfc function and + math module. + """ + def erfcinv(y): + """The inverse function of erfc.""" + if y >= 2: + return -100. + elif y <= 0: + return 100. + zero_point = y < 1 + if not zero_point: + y = 2 - y + t = math.sqrt(-2 * math.log(y / 2.)) + x = -0.70711 * \ + ((2.30753 + t * 0.27061) / (1. + t * (0.99229 + t * 0.04481)) - t) + for i in range(2): + err = erfc(x) - y + x += err / (1.12837916709551257 * math.exp(-(x ** 2)) - x * err) + return x if zero_point else -x + return erfcinv + + +def _gen_ppf(erfc, math=math): + """ppf is the inverse function of cdf. This function generates cdf by the + given erfc and math module. + """ + erfcinv = _gen_erfcinv(erfc, math) + def ppf(x, mu=0, sigma=1): + """The inverse function of cdf.""" + return mu - sigma * math.sqrt(2) * erfcinv(2 * x) + return ppf + + +def erfc(x): + """Complementary error function (via `http://bit.ly/zOLqbc`_)""" + z = abs(x) + t = 1. / (1. + z / 2.) + r = t * math.exp(-z * z - 1.26551223 + t * (1.00002368 + t * ( + 0.37409196 + t * (0.09678418 + t * (-0.18628806 + t * ( + 0.27886807 + t * (-1.13520398 + t * (1.48851587 + t * ( + -0.82215223 + t * 0.17087277 + ))) + ))) + ))) + return 2. - r if x < 0 else r + + +def cdf(x, mu=0, sigma=1): + """Cumulative distribution function""" + return 0.5 * erfc(-(x - mu) / (sigma * math.sqrt(2))) + + +def pdf(x, mu=0, sigma=1): + """Probability density function""" + return (1 / math.sqrt(2 * math.pi) * abs(sigma) * + math.exp(-(((x - mu) / abs(sigma)) ** 2 / 2))) + + +ppf = _gen_ppf(erfc) + + +def choose_backend(backend): + """Returns a tuple containing cdf, pdf, ppf from the chosen backend. + + >>> cdf, pdf, ppf = choose_backend(None) + >>> cdf(-10) + 7.619853263532764e-24 + >>> cdf, pdf, ppf = choose_backend('mpmath') + >>> cdf(-10) + mpf('7.6198530241605255e-24') + + .. versionadded:: 0.3 + + """ + if backend is None: # fallback + return cdf, pdf, ppf + elif backend == 'mpmath': + try: + import mpmath + except ImportError: + raise ImportError('Install "mpmath" to use this backend') + return mpmath.ncdf, mpmath.npdf, _gen_ppf(mpmath.erfc, math=mpmath) + elif backend == 'scipy': + try: + from scipy.stats import norm + except ImportError: + raise ImportError('Install "scipy" to use this backend') + return norm.cdf, norm.pdf, norm.ppf + raise ValueError('%r backend is not defined' % backend) + + +def available_backends(): + """Detects list of available backends. All of defined backends are + ``None`` -- internal implementation, "mpmath", "scipy". + + You can check if the backend is available in the current environment with + this function:: + + if 'mpmath' in available_backends(): + # mpmath can be used in the current environment + setup(backend='mpmath') + + .. versionadded:: 0.3 + + """ + backends = [None] + for backend in ['mpmath', 'scipy']: + try: + __import__(backend) + except ImportError: + continue + backends.append(backend) + return backends diff --git a/data analysis/analysis/trueskill/deprecated.py b/data analysis/analysis/trueskill/deprecated.py new file mode 100644 index 00000000..6378f169 --- /dev/null +++ b/data analysis/analysis/trueskill/deprecated.py @@ -0,0 +1,134 @@ +# -*- coding: utf-8 -*- +""" + trueskill.deprecated + ~~~~~~~~~~~~~~~~~~~~ + + Deprecated features. + + :copyright: (c) 2012-2016 by Heungsub Lee + :license: BSD, see LICENSE for more details. + +""" +from __future__ import absolute_import + +import warnings + +from . import DELTA, expose, global_env, quality_1vs1, rate_1vs1, Rating + + +__all__ = ['transform_ratings', 'match_quality', 'dynamic_draw_probability', + 'ensure_backward_compatibility'] + + +# deprecated functions + + +def transform_ratings(rating_groups, ranks=None, min_delta=DELTA): + return global_env().transform_ratings(rating_groups, ranks, min_delta) + + +def match_quality(rating_groups): + return global_env().match_quality(rating_groups) + + +def dynamic_draw_probability(rating1, rating2, env=None): + """Deprecated. It was an approximation for :func:`quality_1vs1`. + + .. deprecated:: 0.4.1 + Use :func:`quality_1vs1` instead. + + """ + warnings.warn('Use quality_1vs1() instead', DeprecationWarning) + return quality_1vs1(rating1, rating2, env=env) + + +# deprecated methods + + +def addattr(obj, attr, value): + if hasattr(obj, attr): + raise AttributeError('The attribute already exists') + return setattr(obj, attr, value) + + +def ensure_backward_compatibility(TrueSkill, Rating): + addattr(TrueSkill, 'Rating', TrueSkill_Rating) + addattr(TrueSkill, 'transform_ratings', TrueSkill_transform_ratings) + addattr(TrueSkill, 'match_quality', TrueSkill_match_quality) + addattr(TrueSkill, 'rate_1vs1', TrueSkill_rate_1vs1) + addattr(TrueSkill, 'quality_1vs1', TrueSkill_quality_1vs1) + addattr(Rating, 'exposure', Rating_exposure) + + +def TrueSkill_Rating(self, mu=None, sigma=None): + """Deprecated. Used to create a :class:`Rating` object. + + .. deprecated:: 0.2 + Override :meth:`create_rating` instead. + + """ + warnings.warn('Use TrueSkill.create_rating() instead', DeprecationWarning) + return self.create_rating(mu, sigma) + + +def TrueSkill_transform_ratings(self, rating_groups, ranks=None, + min_delta=DELTA): + """Deprecated. Used to rate the given ratings. + + .. deprecated:: 0.2 + Override :meth:`rate` instead. + + """ + warnings.warn('Use TrueSkill.rate() instead', DeprecationWarning) + rating_groups = [(r,) if isinstance(r, Rating) else r + for r in rating_groups] + return self.rate(rating_groups, ranks, min_delta=min_delta) + + +def TrueSkill_match_quality(self, rating_groups): + """Deprecated. Used to calculate a match quality. + + .. deprecated:: 0.2 + Override :meth:`quality` instead. + + """ + warnings.warn('Use TrueSkill.quality() instead', DeprecationWarning) + rating_groups = [(r,) if isinstance(r, Rating) else r + for r in rating_groups] + return self.quality(rating_groups) + + +def TrueSkill_rate_1vs1(self, rating1, rating2, drawn=False, min_delta=DELTA): + """Deprecated. Used to rate just a head-to-haed match. + + .. deprecated:: 0.4 + Use :func:`rate_1vs1` instead. + + """ + warnings.warn('Use global function rate_1vs1() instead', + DeprecationWarning) + return rate_1vs1(rating1, rating2, drawn, min_delta, self) + + +def TrueSkill_quality_1vs1(self, rating1, rating2): + """Deprecated. Used to calculate a match quality for a head-to-haed match. + + .. deprecated:: 0.4 + Use :func:`quality_1vs1` instead. + + """ + warnings.warn('Use global function quality_1vs1() instead', + DeprecationWarning) + return quality_1vs1(rating1, rating2, self) + + +@property +def Rating_exposure(self): + """Deprecated. Used to get a value that will go up on the whole. + + .. deprecated:: 0.4 + Use :meth:`TrueSkill.expose` instead. + + """ + warnings.warn('Use TrueSkill.expose() instead', DeprecationWarning) + return expose(self) diff --git a/data analysis/analysis/trueskill/factorgraph.py b/data analysis/analysis/trueskill/factorgraph.py new file mode 100644 index 00000000..569d522c --- /dev/null +++ b/data analysis/analysis/trueskill/factorgraph.py @@ -0,0 +1,199 @@ +# -*- coding: utf-8 -*- +""" + trueskill.factorgraph + ~~~~~~~~~~~~~~~~~~~~~ + + This module contains nodes for the factor graph of TrueSkill algorithm. + + :copyright: (c) 2012-2016 by Heungsub Lee. + :license: BSD, see LICENSE for more details. + +""" +from __future__ import absolute_import + +import math + +from six.moves import zip + +from .mathematics import Gaussian, inf + + +__all__ = ['Variable', 'PriorFactor', 'LikelihoodFactor', 'SumFactor', + 'TruncateFactor'] + + +class Node(object): + + pass + + +class Variable(Node, Gaussian): + + def __init__(self): + self.messages = {} + super(Variable, self).__init__() + + def set(self, val): + delta = self.delta(val) + self.pi, self.tau = val.pi, val.tau + return delta + + def delta(self, other): + pi_delta = abs(self.pi - other.pi) + if pi_delta == inf: + return 0. + return max(abs(self.tau - other.tau), math.sqrt(pi_delta)) + + def update_message(self, factor, pi=0, tau=0, message=None): + message = message or Gaussian(pi=pi, tau=tau) + old_message, self[factor] = self[factor], message + return self.set(self / old_message * message) + + def update_value(self, factor, pi=0, tau=0, value=None): + value = value or Gaussian(pi=pi, tau=tau) + old_message = self[factor] + self[factor] = value * old_message / self + return self.set(value) + + def __getitem__(self, factor): + return self.messages[factor] + + def __setitem__(self, factor, message): + self.messages[factor] = message + + def __repr__(self): + args = (type(self).__name__, super(Variable, self).__repr__(), + len(self.messages), '' if len(self.messages) == 1 else 's') + return '<%s %s with %d connection%s>' % args + + +class Factor(Node): + + def __init__(self, variables): + self.vars = variables + for var in variables: + var[self] = Gaussian() + + def down(self): + return 0 + + def up(self): + return 0 + + @property + def var(self): + assert len(self.vars) == 1 + return self.vars[0] + + def __repr__(self): + args = (type(self).__name__, len(self.vars), + '' if len(self.vars) == 1 else 's') + return '<%s with %d connection%s>' % args + + +class PriorFactor(Factor): + + def __init__(self, var, val, dynamic=0): + super(PriorFactor, self).__init__([var]) + self.val = val + self.dynamic = dynamic + + def down(self): + sigma = math.sqrt(self.val.sigma ** 2 + self.dynamic ** 2) + value = Gaussian(self.val.mu, sigma) + return self.var.update_value(self, value=value) + + +class LikelihoodFactor(Factor): + + def __init__(self, mean_var, value_var, variance): + super(LikelihoodFactor, self).__init__([mean_var, value_var]) + self.mean = mean_var + self.value = value_var + self.variance = variance + + def calc_a(self, var): + return 1. / (1. + self.variance * var.pi) + + def down(self): + # update value. + msg = self.mean / self.mean[self] + a = self.calc_a(msg) + return self.value.update_message(self, a * msg.pi, a * msg.tau) + + def up(self): + # update mean. + msg = self.value / self.value[self] + a = self.calc_a(msg) + return self.mean.update_message(self, a * msg.pi, a * msg.tau) + + +class SumFactor(Factor): + + def __init__(self, sum_var, term_vars, coeffs): + super(SumFactor, self).__init__([sum_var] + term_vars) + self.sum = sum_var + self.terms = term_vars + self.coeffs = coeffs + + def down(self): + vals = self.terms + msgs = [var[self] for var in vals] + return self.update(self.sum, vals, msgs, self.coeffs) + + def up(self, index=0): + coeff = self.coeffs[index] + coeffs = [] + for x, c in enumerate(self.coeffs): + try: + if x == index: + coeffs.append(1. / coeff) + else: + coeffs.append(-c / coeff) + except ZeroDivisionError: + coeffs.append(0.) + vals = self.terms[:] + vals[index] = self.sum + msgs = [var[self] for var in vals] + return self.update(self.terms[index], vals, msgs, coeffs) + + def update(self, var, vals, msgs, coeffs): + pi_inv = 0 + mu = 0 + for val, msg, coeff in zip(vals, msgs, coeffs): + div = val / msg + mu += coeff * div.mu + if pi_inv == inf: + continue + try: + # numpy.float64 handles floating-point error by different way. + # For example, it can just warn RuntimeWarning on n/0 problem + # instead of throwing ZeroDivisionError. So div.pi, the + # denominator has to be a built-in float. + pi_inv += coeff ** 2 / float(div.pi) + except ZeroDivisionError: + pi_inv = inf + pi = 1. / pi_inv + tau = pi * mu + return var.update_message(self, pi, tau) + + +class TruncateFactor(Factor): + + def __init__(self, var, v_func, w_func, draw_margin): + super(TruncateFactor, self).__init__([var]) + self.v_func = v_func + self.w_func = w_func + self.draw_margin = draw_margin + + def up(self): + val = self.var + msg = self.var[self] + div = val / msg + sqrt_pi = math.sqrt(div.pi) + args = (div.tau / sqrt_pi, self.draw_margin * sqrt_pi) + v = self.v_func(*args) + w = self.w_func(*args) + denom = (1. - w) + pi, tau = div.pi / denom, (div.tau + sqrt_pi * v) / denom + return val.update_value(self, pi, tau) diff --git a/data analysis/analysis/trueskill/mathematics.py b/data analysis/analysis/trueskill/mathematics.py new file mode 100644 index 00000000..61845137 --- /dev/null +++ b/data analysis/analysis/trueskill/mathematics.py @@ -0,0 +1,260 @@ +# -*- coding: utf-8 -*- +""" + trueskill.mathematics + ~~~~~~~~~~~~~~~~~~~~~ + + This module contains basic mathematics functions and objects for TrueSkill + algorithm. If you have not scipy, this module provides the fallback. + + :copyright: (c) 2012-2016 by Heungsub Lee. + :license: BSD, see LICENSE for more details. + +""" +from __future__ import absolute_import + +import copy +import math +try: + from numbers import Number +except ImportError: + Number = (int, long, float, complex) + +from six import iterkeys + + +__all__ = ['Gaussian', 'Matrix', 'inf'] + + +inf = float('inf') + + +class Gaussian(object): + """A model for the normal distribution.""" + + #: Precision, the inverse of the variance. + pi = 0 + #: Precision adjusted mean, the precision multiplied by the mean. + tau = 0 + + def __init__(self, mu=None, sigma=None, pi=0, tau=0): + if mu is not None: + if sigma is None: + raise TypeError('sigma argument is needed') + elif sigma == 0: + raise ValueError('sigma**2 should be greater than 0') + pi = sigma ** -2 + tau = pi * mu + self.pi = pi + self.tau = tau + + @property + def mu(self): + """A property which returns the mean.""" + return self.pi and self.tau / self.pi + + @property + def sigma(self): + """A property which returns the the square root of the variance.""" + return math.sqrt(1 / self.pi) if self.pi else inf + + def __mul__(self, other): + pi, tau = self.pi + other.pi, self.tau + other.tau + return Gaussian(pi=pi, tau=tau) + + def __truediv__(self, other): + pi, tau = self.pi - other.pi, self.tau - other.tau + return Gaussian(pi=pi, tau=tau) + + __div__ = __truediv__ # for Python 2 + + def __eq__(self, other): + return self.pi == other.pi and self.tau == other.tau + + def __lt__(self, other): + return self.mu < other.mu + + def __le__(self, other): + return self.mu <= other.mu + + def __gt__(self, other): + return self.mu > other.mu + + def __ge__(self, other): + return self.mu >= other.mu + + def __repr__(self): + return 'N(mu={:.3f}, sigma={:.3f})'.format(self.mu, self.sigma) + + def _repr_latex_(self): + latex = r'\mathcal{{ N }}( {:.3f}, {:.3f}^2 )'.format(self.mu, self.sigma) + return '$%s$' % latex + + +class Matrix(list): + """A model for matrix.""" + + def __init__(self, src, height=None, width=None): + if callable(src): + f, src = src, {} + size = [height, width] + if not height: + def set_height(height): + size[0] = height + size[0] = set_height + if not width: + def set_width(width): + size[1] = width + size[1] = set_width + try: + for (r, c), val in f(*size): + src[r, c] = val + except TypeError: + raise TypeError('A callable src must return an interable ' + 'which generates a tuple containing ' + 'coordinate and value') + height, width = tuple(size) + if height is None or width is None: + raise TypeError('A callable src must call set_height and ' + 'set_width if the size is non-deterministic') + if isinstance(src, list): + is_number = lambda x: isinstance(x, Number) + unique_col_sizes = set(map(len, src)) + everything_are_number = filter(is_number, sum(src, [])) + if len(unique_col_sizes) != 1 or not everything_are_number: + raise ValueError('src must be a rectangular array of numbers') + two_dimensional_array = src + elif isinstance(src, dict): + if not height or not width: + w = h = 0 + for r, c in iterkeys(src): + if not height: + h = max(h, r + 1) + if not width: + w = max(w, c + 1) + if not height: + height = h + if not width: + width = w + two_dimensional_array = [] + for r in range(height): + row = [] + two_dimensional_array.append(row) + for c in range(width): + row.append(src.get((r, c), 0)) + else: + raise TypeError('src must be a list or dict or callable') + super(Matrix, self).__init__(two_dimensional_array) + + @property + def height(self): + return len(self) + + @property + def width(self): + return len(self[0]) + + def transpose(self): + height, width = self.height, self.width + src = {} + for c in range(width): + for r in range(height): + src[c, r] = self[r][c] + return type(self)(src, height=width, width=height) + + def minor(self, row_n, col_n): + height, width = self.height, self.width + if not (0 <= row_n < height): + raise ValueError('row_n should be between 0 and %d' % height) + elif not (0 <= col_n < width): + raise ValueError('col_n should be between 0 and %d' % width) + two_dimensional_array = [] + for r in range(height): + if r == row_n: + continue + row = [] + two_dimensional_array.append(row) + for c in range(width): + if c == col_n: + continue + row.append(self[r][c]) + return type(self)(two_dimensional_array) + + def determinant(self): + height, width = self.height, self.width + if height != width: + raise ValueError('Only square matrix can calculate a determinant') + tmp, rv = copy.deepcopy(self), 1. + for c in range(width - 1, 0, -1): + pivot, r = max((abs(tmp[r][c]), r) for r in range(c + 1)) + pivot = tmp[r][c] + if not pivot: + return 0. + tmp[r], tmp[c] = tmp[c], tmp[r] + if r != c: + rv = -rv + rv *= pivot + fact = -1. / pivot + for r in range(c): + f = fact * tmp[r][c] + for x in range(c): + tmp[r][x] += f * tmp[c][x] + return rv * tmp[0][0] + + def adjugate(self): + height, width = self.height, self.width + if height != width: + raise ValueError('Only square matrix can be adjugated') + if height == 2: + a, b = self[0][0], self[0][1] + c, d = self[1][0], self[1][1] + return type(self)([[d, -b], [-c, a]]) + src = {} + for r in range(height): + for c in range(width): + sign = -1 if (r + c) % 2 else 1 + src[r, c] = self.minor(r, c).determinant() * sign + return type(self)(src, height, width) + + def inverse(self): + if self.height == self.width == 1: + return type(self)([[1. / self[0][0]]]) + return (1. / self.determinant()) * self.adjugate() + + def __add__(self, other): + height, width = self.height, self.width + if (height, width) != (other.height, other.width): + raise ValueError('Must be same size') + src = {} + for r in range(height): + for c in range(width): + src[r, c] = self[r][c] + other[r][c] + return type(self)(src, height, width) + + def __mul__(self, other): + if self.width != other.height: + raise ValueError('Bad size') + height, width = self.height, other.width + src = {} + for r in range(height): + for c in range(width): + src[r, c] = sum(self[r][x] * other[x][c] + for x in range(self.width)) + return type(self)(src, height, width) + + def __rmul__(self, other): + if not isinstance(other, Number): + raise TypeError('The operand should be a number') + height, width = self.height, self.width + src = {} + for r in range(height): + for c in range(width): + src[r, c] = other * self[r][c] + return type(self)(src, height, width) + + def __repr__(self): + return '{}({})'.format(type(self).__name__, super(Matrix, self).__repr__()) + + def _repr_latex_(self): + rows = [' && '.join(['%.3f' % cell for cell in row]) for row in self] + latex = r'\begin{matrix} %s \end{matrix}' % r'\\'.join(rows) + return '$%s$' % latex