X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=train.pl;h=9238b61cb73ef1e6945c352a46f5f04f257906e3;hb=50ff23153d61665500e9919a51cb34c4774d492c;hp=67b2b8d737e49f0dbde6f004923cc8f6af7b5c83;hpb=72a2eedd021d38d6bb0786033c6f100062363789;p=wloh

diff --git a/train.pl b/train.pl
index 67b2b8d..9238b61 100755
--- a/train.pl
+++ b/train.pl
@@ -11,30 +11,14 @@ my $dbh = DBI->connect($config::local_connstr, $config::local_username, $config:
 $dbh->{AutoCommit} = 0;
 $dbh->{RaiseError} = 1;
 
+$dbh->do('SET client_min_messages TO WARNING');
+
 # Find last completely done season 
 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');
 my $last_season = $ref->{'sesong'};
 
-my $tmpnam = POSIX::tmpnam();
-open DATA, ">", $tmpnam
-	or die "$tmpnam: $!";
-
-# Fetch name (ID) list
-my $q = $dbh->prepare('SELECT DISTINCT id FROM fotballdeltagere');
-$q->execute();
-my @ids = ();
-while (my $ref = $q->fetchrow_hashref) {
-	my $id = $ref->{'id'};
-	push @ids, $id;
-}
-	
-print DATA scalar @ids, "\n";
-for my $id (@ids) {
-	print DATA $id, "\n";
-}
-
 # Fetch games
-$q = $dbh->prepare('
+my $q = $dbh->prepare('
 SELECT
   deltager1.id as p1, deltager2.id as p2, maalfor, maalmot, least(pow(2.0, (sesong - ? + 3) / 3.0), 1.0) AS vekt
 FROM
@@ -46,24 +30,59 @@ WHERE deltager1.Nr > deltager2.nr
 ');
 $q->execute($last_season);
 
+my @games = ();
+my %ids = ();
+
 while (my $ref = $q->fetchrow_hashref) {
 	next if ($ref->{'maalfor'} == 150 && $ref->{'maalmot'} == 0);
 	next if ($ref->{'maalfor'} == 0 && $ref->{'maalmot'} == 150);
 	next if ($ref->{'maalfor'} == 150 && $ref->{'maalmot'} == 150);
-	printf DATA "%d %d %d %d %f\n", $ref->{'p1'}, $ref->{'p2'}, $ref->{'maalfor'}, $ref->{'maalmot'}, $ref->{'vekt'};
+	push @games, { %$ref };
+	$ids{$ref->{'p1'}} = 1;
+	$ids{$ref->{'p2'}} = 1;
 }
 
+# Output to file
+my $tmpnam = POSIX::tmpnam();
+open DATA, ">", $tmpnam
+	or die "$tmpnam: $!";
+
+printf DATA "%d\n", scalar keys %ids;
+for my $id (keys %ids) {
+	printf DATA "%d\n", $id;
+}
+for my $ref (@games) {
+	printf DATA "%d %d %d %d %f\n", $ref->{'p1'}, $ref->{'p2'}, $ref->{'maalfor'}, $ref->{'maalmot'}, $ref->{'vekt'};
+}
 close DATA;
 
-$dbh->do('DELETE FROM ratings');
-my $iq = $dbh->prepare('INSERT INTO ratings ( id, rating, rating_stddev ) VALUES (?, ?, ?)');
+my @ratings = ();
+my @covariances = ();
 
 open RATINGS, "$config::base_dir/bayeswf < $tmpnam |"
 	or die "bayeswf: $!";
 while (<RATINGS>) {
-	/(.*) (.*) (.*)/ or next;
-	$iq->execute($3, $1, $2);
+	chomp;
+	my @x = split;
+	if ($x[0] eq 'covariance') {
+		push @covariances, (join("\t", @x[1..3]));
+	} else {
+		push @ratings, ($x[2] . "\t" . $x[0] . "\t" . $x[1]);
+	}
 }
 
+$dbh->do('TRUNCATE ratings');
+$dbh->do('COPY ratings ( id, rating, rating_stddev ) FROM STDIN');
+$dbh->pg_putcopydata(join("\n", @ratings));
+$dbh->pg_putcopyend();
+
+$dbh->do('CREATE TABLE new_covariance ( player1 smallint NOT NULL, player2 smallint NOT NULL, cov float NOT NULL )');
+$dbh->do('COPY new_covariance ( player1, player2, cov ) FROM STDIN');
+$dbh->pg_putcopydata(join("\n", @covariances));
+$dbh->pg_putcopyend();
+$dbh->do('ALTER TABLE new_covariance ADD PRIMARY KEY ( player1, player2 );');
+$dbh->do('DROP TABLE IF EXISTS covariance');
+$dbh->do('ALTER TABLE new_covariance RENAME TO covariance');
+
 $dbh->commit;
 unlink($tmpnam);