skab 0.1.0 → 0.1.2

data/lib/skab/models/binomial.rb

@@ -7,6 +7,7 @@ module Skab
         @a_success = args.shift.to_i
         @b_trials = args.shift.to_i
         @b_success = args.shift.to_i
+        @fact = { }
       end
 
       def distribution
@@ -67,17 +68,13 @@ skab [output] binomial [trials_a] [successes_a] [trials_b] [successes_b]
       end
 
       def binomial_coef(n, k)
-        fact(n) / (fact(k) * fact(n - k))
+        factorial(n) / (factorial(k) * factorial(n - k))
       end
 
-      def fact(n)
-        f = 1
-        (1..n).each do |i|
-          f *= i
-        end
-        f
+      def factorial(n)
+        return @fact[n] if @fact[n]
+        @fact[n] = (n > 1) ? n * factorial(n - 1) : 1
       end
-
     end
   end
 end

data/lib/skab/models/poisson.rb

@@ -5,6 +5,7 @@ module Skab
       def initialize(args)
         @a = args.shift.to_i
         @b = args.shift.to_i
+        @fact = { }
       end
 
       def distribution
@@ -55,13 +56,9 @@ skab [output] poisson [a] [b]
       end
 
       def factorial(n)
-        f = 1
-        (1..n).each do |i|
-          f *= i
-        end
-        f
+        return @fact[n] if @fact[n]
+        @fact[n] = (n > 1) ? n * fact(n - 1) : 1
       end
-
     end
   end
 end

data/lib/skab/output.rb

@@ -2,6 +2,8 @@ module Skab
   require ROOT + '/skab/output/distribution'
   require ROOT + '/skab/output/differential'
   require ROOT + '/skab/output/summary'
+  require ROOT + '/skab/output/gnuplot_distribution'
+  require ROOT + '/skab/output/gnuplot_differential'
 
   module Output
     def self.from_name(name)
@@ -12,11 +14,21 @@ module Skab
           Distribution
         when 'summary'
           Summary
+        when 'gnuplot_distribution'
+          GnuplotDistribution
+        when 'gnuplot_differential'
+          GnuplotDifferential
       end
     end
 
     def self.output_names
-      ['distribution', 'differential', 'summary']
+      [
+        'distribution', 
+        'differential', 
+        'summary', 
+        'gnuplot_distribution',
+        'gnuplot_differential'
+      ]
     end
 
     def self.help

data/lib/skab/output/gnuplot_differential.rb

@@ -0,0 +1,23 @@
+module Skab
+  module Output
+    class GnuplotDifferential
+      def initialize(out)
+        @out = out
+      end
+
+      def output(model)
+        @out.puts "set encoding utf8"
+        @out.puts "set title"
+        @out.puts "set key outside"
+        @out.puts "set title \"Difference distribution\""
+
+        @out.puts "set style fill transparent solid 0.5 border"
+
+        @out.puts "plot '-' using 1:2 with filledcurve lc rgb 'red' title \"Pr(\316\264)\""
+        model.differential.sort.each do |k, v|
+          @out.puts " #{k} #{v}"
+        end
+      end
+    end
+  end
+end

data/lib/skab/output/gnuplot_distribution.rb

@@ -0,0 +1,29 @@
+module Skab
+  module Output
+    class GnuplotDistribution
+      def initialize(out)
+        @out = out
+      end
+
+      def output(model)
+        @out.puts "set title"
+        @out.puts "set key outside"
+        @out.puts "set title \"Probability density distribution\""
+
+        @out.puts "set style fill transparent solid 0.5 border"
+
+        interval = [model.distribution.first[0], model.distribution.last[0]]
+        @out.puts "plot '-' using 1:2 with filledcurve lc rgb 'red' title 'Group A', \\"
+        @out.puts "\t'' using 1:2 with filledcurve lc rgb 'blue' title 'Group B'"
+        model.distribution.each do |d|
+          @out.puts " #{d[0]} #{d[1]}"
+        end
+        @out.puts "e"
+        model.distribution.each do |d|
+          @out.puts " #{d[0]} #{d[2]}"
+        end
+        @out.puts "e"
+      end
+    end
+  end
+end

metadata

@@ -1,64 +1,60 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: skab
-version: !ruby/object:Gem::Version 
+version: !ruby/object:Gem::Version
+  version: 0.1.2
   prerelease: 
-  version: 0.1.0
 platform: ruby
-authors: 
+authors:
 - Vivien Barousse
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2012-10-10 00:00:00 Z
+date: 2013-05-30 00:00:00.000000000 Z
 dependencies: []
-
 description: 
 email: vivien@songkick.com
-executables: 
+executables:
 - skab
 extensions: []
-
-extra_rdoc_files: 
+extra_rdoc_files:
 - README.rdoc
-files: 
+files:
 - README.rdoc
 - bin/skab
 - lib/skab.rb
-- lib/skab/output.rb
-- lib/skab/models/binomial.rb
-- lib/skab/models/poisson.rb
 - lib/skab/output/summary.rb
 - lib/skab/output/distribution.rb
 - lib/skab/output/differential.rb
+- lib/skab/output/gnuplot_distribution.rb
+- lib/skab/output/gnuplot_differential.rb
+- lib/skab/output.rb
+- lib/skab/models/binomial.rb
+- lib/skab/models/poisson.rb
 - lib/skab/models.rb
 homepage: http://github.com/songkick/skab
 licenses: []
-
 post_install_message: 
-rdoc_options: 
+rdoc_options:
 - --main
 - README.rdoc
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
+required_ruby_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      version: "0"
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
 requirements: []
-
 rubyforge_project: 
-rubygems_version: 1.8.21
+rubygems_version: 1.8.25
 signing_key: 
 specification_version: 3
 summary: A/B testing statistical analysis utility
 test_files: []
-