git.sesse.net Git - foosball/blob - recalc-single-result.pl

   1 #! /usr/bin/perl
   2 use strict;
   3 use warnings;
   4 use DBI;
   5 use CGI;
   6 use CGI::Carp qw(fatalsToBrowser);
   7 require 'foosball.pm';
   8 use foosrank;
   9 no warnings 'once';
  10
  11 my $dbh = foosball::db_connect();
  12 $dbh->{AutoCommit} = 0;
  13
  14 my $q = $dbh->prepare('select *,extract(epoch from gametime) as eptime from single_results order by gametime');
  15 $q->execute;
  16 my $games = $q->fetchall_arrayref({});
  17
  18 my $dryrun = 0;
  19 my @parms = ();
  20 if (defined($ARGV[0])) {
  21         $dryrun = 1;
  22         for my $arg (@ARGV) {
  23                 my ($rd, $c) = split /,/, $arg;
  24                 push @parms, [ $rd, $c ];
  25         }
  26 } else {
  27         @parms = ([ $foosball::initial_rd, $foosball::c ]);
  28 }
  29
  30 for my $parm (@parms) {
  31         $foosball::initial_rd = $parm->[0];
  32         $foosball::c = $parm->[1];
  33
  34         $dbh->do('delete from single_rating') unless $dryrun;
  35
  36         my %ratings = ();
  37
  38         # Combined log-likelihood
  39         my $cll = 0.0;
  40
  41         for my $ref (@$games) {
  42                 for my $user (($ref->{'username1'}, $ref->{'username2'})) {
  43                         if (!exists($ratings{$user})) {
  44                                 $ratings{$user} = [ $foosball::initial_rating, $foosball::initial_rd, 0 ];
  45                         }
  46                 }
  47
  48                 my $rating1 = $ratings{$ref->{'username1'}}->[0];
  49                 my $rd1 = $ratings{$ref->{'username1'}}->[1];
  50                 my $age1 = $ref->{'eptime'} - $ratings{$ref->{'username1'}}->[2];
  51
  52                 my $rating2 = $ratings{$ref->{'username2'}}->[0];
  53                 my $rd2 = $ratings{$ref->{'username2'}}->[1];
  54                 my $age2 = $ref->{'eptime'} - $ratings{$ref->{'username2'}}->[2];
  55
  56                 my $score1 = $ref->{'score1'};
  57                 my $score2 = $ref->{'score2'};
  58
  59                 $rd1 = foosball::apply_aging($rd1, $age1 / 86400.0);
  60                 $rd2 = foosball::apply_aging($rd2, $age2 / 86400.0);
  61
  62                 my ($newr1, $newrd1, $likelihood) = foosrank::compute_new_rating($rating1, $rd1, $rating2, $rd2, $score1, $score2);
  63                 my ($newr2, $newrd2) = foosrank::compute_new_rating($rating2, $rd2, $rating1, $rd1, $score2, $score1);
  64
  65                 $cll += log($likelihood);
  66
  67                 unless ($dryrun) {
  68                         printf("%-10s - %-10s: %u - %u, new ratings %u/%u %u/%u [$likelihood]\n",
  69                                 $ref->{'username1'}, $ref->{'username2'}, $ref->{'score1'},
  70                                 $ref->{'score2'}, $newr1, $newrd1, $newr2, $newrd2);
  71                         $dbh->do('insert into single_rating values (?,?,?,?,?)', undef,
  72                                 $ref->{'username1'}, $ref->{'gametime'}, $newr1, $newrd1, $newr1-$rating1);
  73                         $dbh->do('insert into single_rating values (?,?,?,?,?)', undef,
  74                                 $ref->{'username2'}, $ref->{'gametime'}, $newr2, $newrd2, $newr2-$rating2);
  75                 }
  76
  77                 $ratings{$ref->{'username1'}} = [ $newr1, $newrd1, $ref->{'eptime'} ];
  78                 $ratings{$ref->{'username2'}} = [ $newr2, $newrd2, $ref->{'eptime'} ];
  79         }
  80
  81         $dbh->commit unless $dryrun;
  82
  83         if ($dryrun) {
  84                 printf "%f %f %f\n", $parm->[0], $parm->[1], -$cll;
  85         } else {
  86                 printf "\nCombined negative log-likelihood (smaller value means a better model): %f\n",
  87                         -$cll;
  88         }
  89 }
  90 $dbh->disconnect;