Sha256: 5a77c057e0ced2edfc919c8f77261b306062996afc5f0042d7f610c387d2432a

Contents?: true

Size: 1.28 KB

Versions: 13

Compression:

Stored size: 1.28 KB

Contents

module Statsample
  module Test
    module ChiSquare
      class WithMatrix
        attr_reader :df
        attr_reader :value
        def initialize(observed, expected=nil)
          @observed=observed
          @expected=expected or calculate_expected
          raise "Observed size!=expected size" if @observed.row_size!=@expected.row_size or @observed.column_size!=@expected.column_size
          @df=(@observed.row_size-1)*(@observed.column_size-1)
          @value=compute_chi
        end
        def calculate_expected
          sum=@observed.total_sum
          @expected=Matrix.rows( @observed.row_size.times.map {|i|
            @observed.column_size.times.map {|j|
              (@observed.row_sum[i].quo(sum) * @observed.column_sum[j].quo(sum))*sum
            }
          })          
        end
        def to_f
          @value
        end
        def chi_square
          @value
        end
        def probability
          1-Distribution::ChiSquare.cdf(@value.to_f,@df)
        end
        def compute_chi
            sum=0
            (0...@observed.row_size).each {|i|
              (0...@observed.column_size).each {|j|
              sum+=((@observed[i, j] - @expected[i,j])**2).quo(@expected[i,j])
              }
            }
            sum
        end
      end
    end
  end
end

Version data entries

13 entries across 13 versions & 1 rubygems

Version Path
statsample-2.0.1 lib/statsample/test/chisquare.rb
statsample-2.0.0 lib/statsample/test/chisquare.rb
statsample-1.5.0 lib/statsample/test/chisquare.rb
statsample-1.4.3 lib/statsample/test/chisquare.rb
statsample-1.4.2 lib/statsample/test/chisquare.rb
statsample-1.4.1 lib/statsample/test/chisquare.rb
statsample-1.4.0 lib/statsample/test/chisquare.rb
statsample-1.3.1 lib/statsample/test/chisquare.rb
statsample-1.3.0 lib/statsample/test/chisquare.rb
statsample-1.2.0 lib/statsample/test/chisquare.rb
statsample-1.1.0 lib/statsample/test/chisquare.rb
statsample-1.0.1 lib/statsample/test/chisquare.rb
statsample-1.0.0 lib/statsample/test/chisquare.rb