magarena/exp/whr.rb

24 lines
775 B
Ruby
Executable File

#!/usr/bin/env ruby
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(' ')
@whr.create_game(tokens[0], tokens[1], tokens[2], 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