trueskill 0.1.0

data/lib/saulabs/trueskill/layers/team_difference_comparision.rb

@@ -0,0 +1,27 @@
+module Saulabs
+  module TrueSkill
+    module Layers
+    
+      class TeamDifferenceComparision < Base
+        
+        def initialize(graph, ranks)
+          super(graph)
+          @ranks = ranks
+          @epsilon = graph.draw_margin
+        end
+        
+        def build
+          (0..@input.size-1).each do |i|
+            if @ranks[i] == @ranks[i+1]
+              @factors << Factors::Within.new(@epsilon, @input[i][0])
+            else
+              @factors << Factors::GreaterThan.new(@epsilon, @input[i][0])
+            end
+          end
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/saulabs/trueskill/layers/team_performance_differences.rb

@@ -0,0 +1,22 @@
+module Saulabs
+  module TrueSkill
+    module Layers
+    
+      class TeamPerformanceDifferences < Base
+        
+        def initialize(graph)
+          super(graph)
+        end
+        
+        def build
+          (0..@input.size-2).each do |i|
+            variable = Gauss::Distribution.new
+            @factors << Factors::WeightedSum.new(variable, [@input[i][0], @input[i+1][0]], [1.0, -1.0])
+            @output << [variable]
+          end
+        end
+      end
+      
+    end
+  end
+end

data/lib/saulabs/trueskill/rating.rb

@@ -0,0 +1,18 @@
+module Saulabs
+  module TrueSkill
+    
+    class Rating < Gauss::Distribution
+      
+      attr_reader :activity, :tau, :tau_squared
+      
+      def initialize(mean, deviation, activity = 1.0, tau = 25/300.0)
+        super(mean, deviation)
+        @activity = activity
+        @tau = tau
+        @tau_squared = @tau**2
+      end
+      
+    end
+    
+  end
+end

data/lib/saulabs/trueskill/schedules/base.rb

@@ -0,0 +1,15 @@
+module Saulabs
+  module TrueSkill
+    module Schedules
+      
+      class Base
+        
+        def visit(depth = -1, max_depth = 0)
+          raise "Abstract method Schedules::Base#visit(depth, max_depth) called"
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/saulabs/trueskill/schedules/loop.rb

@@ -0,0 +1,26 @@
+module Saulabs
+  module TrueSkill
+    module Schedules
+      
+      class Loop < Base
+        
+        def initialize(schedule, max_delta)
+          @schedule = schedule
+          @max_delta = max_delta
+        end
+        
+        def visit(depth = -1, max_depth = 0)
+          iterations = 1
+          delta = @schedule.visit(depth + 1, max_depth)
+          while delta > @max_delta
+            delta = @schedule.visit(depth + 1, max_depth)
+            iterations += 1
+          end
+          delta
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/saulabs/trueskill/schedules/sequence.rb

@@ -0,0 +1,23 @@
+module Saulabs
+  module TrueSkill
+    module Schedules
+      
+      class Sequence < Base
+        
+        def initialize(schedules)
+          @schedules = schedules
+        end
+        
+        def visit(depth = -1, max_depth = 0)
+          max_delta = 0
+          @schedules.each do |schedule|
+            max_delta = [schedule.visit(depth + 1, max_depth), max_delta].max
+          end
+          max_delta
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/saulabs/trueskill/schedules/step.rb

@@ -0,0 +1,20 @@
+module Saulabs
+  module TrueSkill
+    module Schedules
+      
+      class Step < Base
+        
+        def initialize(factor, index)
+          @factor = factor
+          @index = index
+        end
+        
+        def visit(depth = -1, max_depth = 0)
+          @factor.update_message_at(@index)
+        end          
+        
+      end
+      
+    end
+  end
+end

data/spec/saulabs/gauss/distribution_spec.rb

@@ -0,0 +1,162 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+describe Gauss::Distribution, "#initialize" do
+
+  it "should set the mean to 10.1" do
+    Gauss::Distribution.new(10.1, 0.4).mean.should == 10.1
+  end
+  
+  it "should set the deviation to 0.4" do
+    Gauss::Distribution.new(10.1, 0.4).deviation.should == 0.4
+  end
+  
+  it "should set the mean to 0.0 if the given mean is not finite" do
+    Gauss::Distribution.new(1 / 0.0, 0.4).mean.should == 0.0
+  end
+  
+  it "should set the deviation to 0.0 if the given deviation is not finite" do
+    Gauss::Distribution.new(10.1, 1 / 0.0).deviation.should == 0.0
+  end
+  
+end
+
+describe Gauss::Distribution, "#with_deviation" do
+  
+  before :each do 
+    @dist = Gauss::Distribution.with_deviation(25.0, 8.333333)
+  end
+
+  it "should have a default mean value of 25.0" do
+    @dist.mean.should == 25.0
+  end
+
+  it "should have a default deviation of 8.333333" do
+    @dist.deviation.should be_close(8.333333, 0.000001)
+  end
+
+  it "should set the variance to 69.444438" do
+    @dist.variance.should be_close(69.4444, 0.0001)
+  end
+
+  it "should set the precision to 0.0144" do
+    @dist.precision.should be_close(0.0144, 0.0001)
+  end
+
+  it "should set the precision_mean to 0.36" do
+    @dist.precision_mean.should be_close(0.36, 0.0001)
+  end
+  
+end
+  
+describe Gauss::Distribution, "#with_precision" do
+  
+  before :each do 
+    @dist = Gauss::Distribution.with_precision(0.36, 0.0144)
+  end
+
+  it "should have a default mean value of 25.0" do
+    @dist.mean.should == 25.0
+  end
+
+  it "should have a default deviation of 8.333333" do
+    @dist.deviation.should be_close(8.333333, 0.000001)
+  end
+
+  it "should set the variance to 69.444438" do
+    @dist.variance.should be_close(69.4444, 0.0001)
+  end
+
+  it "should set the precision to 0.0144" do
+    @dist.precision.should be_close(0.0144, 0.0001)
+  end
+
+  it "should set the precision_mean to 0.36" do
+    @dist.precision_mean.should be_close(0.36, 0.0001)
+  end
+  
+end
+
+describe Gauss::Distribution, "absolute difference (-)" do
+  
+  before :each do 
+    @dist = Gauss::Distribution.with_deviation(25.0, 8.333333)
+  end
+  
+  it "should be 0.0 for the same distribution" do
+    (@dist - @dist).should == 0.0
+  end
+  
+  it "should equal the precision mean if the 0-distribution is subtracted" do
+    (@dist - Gauss::Distribution.new).should == @dist.precision_mean
+  end
+  
+  it "should be 130.399408 for (22, 0.4) - (12, 1.3)" do
+    (Gauss::Distribution.new(22, 0.4) - Gauss::Distribution.new(12, 1.3)).should be_close(130.399408, tolerance)
+  end
+  
+end
+
+describe Gauss::Distribution, "#value_at" do
+  
+  it "should have a value of 0.073654 for x = 2" do
+    Gauss::Distribution.new(4,5).value_at(2).should be_close(0.073654, tolerance)
+  end
+  
+end
+
+describe Gauss::Distribution, "multiplication (*)" do
+  
+  it "should have a mean of 0.2" do
+    (Gauss::Distribution.new(0,1) * Gauss::Distribution.new(2,3)).mean.should be_close(0.2, 0.00001)
+  end
+  
+  it "should have a deviation of 3.0 / Math.sqrt(10)" do
+    (Gauss::Distribution.new(0,1) * Gauss::Distribution.new(2,3)).deviation.should be_close(3.0 / Math.sqrt(10), 0.00001)
+  end
+  
+end
+
+describe Gauss::Distribution, "#log_product_normalization" do
+  
+  it "should have calculate -3.0979981" do
+    lp = Gauss::Distribution.log_product_normalization(Gauss::Distribution.new(4,5), Gauss::Distribution.new(6,7))
+    lp.should be_close(-3.0979981, 0.000001)
+  end
+  
+end
+
+describe Gauss::Distribution, "functions" do
+  
+  describe 'value = 0.27' do
+    
+    it "#cumulative_distribution_function should return 0.6064198" do
+      Gauss::Distribution.cumulative_distribution_function(0.27).should be_close(0.6064198, 0.00001)
+      Gauss::Distribution.cdf(2.0).should be_close(0.9772498, 0.00001)
+    end
+    
+    it "#probability_density_function should return 0.384662" do
+      Gauss::Distribution.probability_density_function(0.27).should be_close(0.384662, 0.0001)
+    end
+    
+    it "#quantile_function should return -0.62941" do
+      Gauss::Distribution.quantile_function(0.27).should be_close(-0.62941, 0.00001)
+      Gauss::Distribution.quantile_function(0.9).should be_close(1.281551, tolerance)
+    end
+    
+  end
+  
+end
+
+describe Gauss::Distribution, "#replace" do
+  
+  before :each do 
+    @dist1 = Gauss::Distribution.with_deviation(25.0, 8.333333)
+    @dist2 = Gauss::Distribution.with_deviation(9.0, 4)
+  end
+  
+  it "should be equal to the replaced distribution" do
+    @dist1.replace(@dist2)
+    @dist1.should == @dist2
+  end
+  
+end

data/spec/saulabs/gauss/truncated_correction_spec.rb

@@ -0,0 +1,41 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+describe Gauss::TruncatedCorrection do
+
+  describe "#w_within_margin" do
+
+    it "should return 0.970397 for (0.2, 0.3)" do
+      Gauss::TruncatedCorrection.w_within_margin(0.2, 0.3).should be_close(0.970397, tolerance)
+      Gauss::TruncatedCorrection.w_within_margin(0.1, 0.03).should be_close(0.9997, tolerance)
+    end
+
+  end
+
+  describe "#v_within_margin" do
+
+    it "should return -0.194073 for (0.2, 0.3)" do
+      Gauss::TruncatedCorrection.v_within_margin(0.2, 0.3).should be_close(-0.194073, tolerance)
+      Gauss::TruncatedCorrection.v_within_margin(0.1, 0.03).should be_close(-0.09997, tolerance)
+    end
+
+  end
+  
+  describe "#w_exceeds_margin" do
+  
+    it "should return 0.657847 for (0.2, 0.3)" do
+      Gauss::TruncatedCorrection.w_exceeds_margin(0.2, 0.3).should be_close(0.657847, tolerance)
+      Gauss::TruncatedCorrection.w_exceeds_margin(0.1, 0.03).should be_close(0.621078, tolerance)
+    end
+  
+  end
+  
+  describe "#v_exceeds_margin" do
+  
+    it "should return 0.8626174 for (0.2, 0.3)" do
+      Gauss::TruncatedCorrection.v_exceeds_margin(0.2, 0.3).should be_close(0.8626174, tolerance)
+      Gauss::TruncatedCorrection.v_exceeds_margin(0.1, 0.03).should be_close(0.753861, tolerance)
+    end
+  
+  end
+
+end

data/spec/saulabs/trueskill/factor_graph_spec.rb

@@ -0,0 +1,29 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+describe Saulabs::TrueSkill::FactorGraph do
+  
+  before :each do
+    @teams = create_teams
+    @graph = TrueSkill::FactorGraph.new(@teams, [1,2,3])
+  end
+  
+  describe "#evaluate" do
+    
+    it "should do something" do
+      result = @graph.evaluate
+      puts "[#{result.last.flatten.map(&:to_s).join(', ')}]<br>"
+      result.last[0][0].mean.should be_close(29.61452, tolerance)
+      result.last[0][0].deviation.should be_close(3.5036, tolerance)
+    end
+    
+  end
+  
+  describe "#draw_margin" do
+    
+    it "should be " do
+      @graph.draw_margin.should be_close(-0.998291, tolerance)
+    end
+    
+  end
+  
+end

data/spec/saulabs/trueskill/factors/greater_than_spec.rb

@@ -0,0 +1,26 @@
+require File.dirname(__FILE__) + '/../../../spec_helper'
+
+describe TrueSkill::Factors::GreaterThan do
+  
+  before :each do
+    @variable = Gauss::Distribution.new(0.1, 1.1)
+    @factor = TrueSkill::Factors::GreaterThan.new(0.1, @variable)
+  end
+  
+  describe "#update_message_at" do
+    
+    it "should return a difference of 2.1409" do
+      @factor.update_message_at(0).should be_close(2.1409, tolerance)
+    end
+  
+  end
+  
+  describe "#log_normalization" do
+    
+    it "should be -0.69314" do
+      @factor.log_normalization.should be_close(-0.69314, tolerance)
+    end
+  
+  end
+  
+end

data/spec/saulabs/trueskill/factors/likelihood_spec.rb

@@ -0,0 +1,32 @@
+require File.dirname(__FILE__) + '/../../../spec_helper'
+
+describe TrueSkill::Factors::Likelihood do
+  
+  before :each do
+    @variable1 = Gauss::Distribution.new(26, 1.1)
+    @variable2 = Gauss::Distribution.new
+    @factor = TrueSkill::Factors::Likelihood.new(30, @variable1, @variable2)
+  end
+  
+  describe "#update_message_at" do
+    
+    it "should return a difference of 0.0" do
+      @factor.update_message_at(0).should be_close(0.0, tolerance)
+    end
+    
+    it "should return a difference of 0.833066 for the second message" do
+      @factor.update_message_at(0)
+      @factor.update_message_at(1).should be_close(0.833066, tolerance)
+    end
+  
+  end
+  
+  describe "#log_normalization" do
+    
+    it "should be 0.0" do
+      @factor.log_normalization.should == 0.0
+    end
+  
+  end
+  
+end