magarena/exp/whr.rb

34 lines
1.0 KiB
Ruby
Executable File

#!/usr/bin/env ruby
# read from stdin lines in the following format:
# # <player0 name> <player1 name> <winner>
# # where <winner> is 0 if player0 wins and 1 if player1 wins
require 'whole_history_rating'
# w = 0 for static ratings aka bayeselo
@whr = WholeHistoryRating::Base.new(:w2 => 0)
# WholeHistoryRating::Base#create_game arguments: black player name, white player name, winner, day number, handicap
# Handicap should generally be less than 500 elo
ARGF.each do |line|
tokens = line.split(' ')
winner =
if (tokens[2] == "0")
"B"
else (tokens[2] == "1")
"W"
end
@whr.create_game(tokens[0], tokens[1], winner, 0, 0)
end
# Iterate the WHR algorithm towards convergence with more players/games, more iterations are needed.
@whr.iterate(1000)
players = @whr.players.values.select {|p| p.days.count > 0}
players.sort_by { |p| p.days.last.gamma }.each_with_index do |p,idx|
if p.days.count > 0
puts "#{p.name}\t#{p.days[0].gamma}\t#{p.days[0].uncertainty}"
end
end