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