git.sesse.net Git - wloh/blob - train.pl

   1 #! /usr/bin/perl
   2 use DBI;
   3 use strict;
   4 use warnings;
   5 no warnings qw(once);
   6 use POSIX;
   7 require './config.pm';
   8
   9 my $dbh = DBI->connect($config::local_connstr, $config::local_username, $config::local_password)
  10         or die "connect: " . $DBI::errstr;
  11 $dbh->{AutoCommit} = 0;
  12 $dbh->{RaiseError} = 1;
  13
  14 # Find last completely done season
  15 my $ref = $dbh->selectrow_hashref('SELECT sesong FROM fotballserier GROUP BY sesong HAVING COUNT(*)=COUNT(avgjort=1 OR NULL) ORDER BY sesong DESC LIMIT 1');
  16 my $last_season = $ref->{'sesong'};
  17
  18 my $tmpnam = POSIX::tmpnam();
  19 open DATA, ">", $tmpnam
  20         or die "$tmpnam: $!";
  21
  22 # Fetch name (ID) list
  23 my $q = $dbh->prepare('SELECT DISTINCT id FROM fotballdeltagere');
  24 $q->execute();
  25 my @ids = ();
  26 while (my $ref = $q->fetchrow_hashref) {
  27         my $id = $ref->{'id'};
  28         push @ids, $id;
  29 }
  30
  31 print DATA scalar @ids, "\n";
  32 for my $id (@ids) {
  33         print DATA $id, "\n";
  34 }
  35
  36 # Fetch games
  37 $q = $dbh->prepare('
  38 SELECT
  39   deltager1.id as p1, deltager2.id as p2, maalfor, maalmot, least(pow(2.0, (sesong - ? + 3) / 3.0), 1.0) AS vekt
  40 FROM
  41   Fotballresultater resultater
  42   JOIN Fotballdeltagere deltager1 ON resultater.Lagrecno=deltager1.Nr AND resultater.Serie=deltager1.Serie
  43   JOIN Fotballdeltagere deltager2 ON resultater.Motstander=deltager2.Nr AND resultater.Serie=deltager2.Serie
  44   JOIN Fotballserier serier on resultater.Serie=serier.Nr
  45 WHERE deltager1.Nr > deltager2.nr
  46 ');
  47 $q->execute($last_season);
  48
  49 while (my $ref = $q->fetchrow_hashref) {
  50         next if ($ref->{'maalfor'} == 150 && $ref->{'maalmot'} == 0);
  51         next if ($ref->{'maalfor'} == 0 && $ref->{'maalmot'} == 150);
  52         next if ($ref->{'maalfor'} == 150 && $ref->{'maalmot'} == 150);
  53         printf DATA "%d %d %d %d %f\n", $ref->{'p1'}, $ref->{'p2'}, $ref->{'maalfor'}, $ref->{'maalmot'}, $ref->{'vekt'};
  54 }
  55
  56 close DATA;
  57
  58 $dbh->do('DELETE FROM ratings');
  59 my $iq = $dbh->prepare('INSERT INTO ratings ( id, rating, rating_stddev ) VALUES (?, ?, ?)');
  60
  61 open RATINGS, "$config::base_dir/bayeswf < $tmpnam |"
  62         or die "bayeswf: $!";
  63 while (<RATINGS>) {
  64         /(.*) (.*) (.*)/ or next;
  65         $iq->execute($3, $1, $2);
  66 }
  67
  68 $dbh->commit;
  69 unlink($tmpnam);