abba 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 88201a4cd5a8b7ea2eb7a28e31b5a5cd7ac86fc9
+  data.tar.gz: c1f24a23ff75f646ee43f7da4368e807c40f7a73
+SHA512:
+  metadata.gz: 76adccef44a5bb5697166d449d2a8bdd285c3bbada18dce1b884d61b84841ab30a3e22e9f11d0ec47338709ac804972277468f54f8e048ca64fd8455c1f0390d
+  data.tar.gz: b8922e2fa9e9678062e8d709b7171a482bd9e225381fec156a720fc9ab19ed64b01b17c971b8b9a8fa44dbf182ac2d96bbde28c27515489d0a55c8c911e054da

data/lib/abba.rb

@@ -0,0 +1,6 @@
+module ABBA
+end
+
+require 'abba/bayes'
+require 'abba/math'
+require 'abba/ab_test'

data/lib/abba/ab_test.rb

@@ -0,0 +1,84 @@
+module ABBA
+  class ABTest
+    def initialize(variant_a:, variant_b:, test: Bayes::BinaryOutcomeTest)
+      @variant_a = ABTest::Variant.new(variant_a)
+      @variant_b = ABTest::Variant.new(variant_b)
+      @test = test
+    end
+
+    def choose_a?
+      @test.new(@variant_a, @variant_b).call
+    end
+
+    def choose_b?
+      @test.new(@variant_b, @variant_a).call
+    end
+
+    def probabilities
+      p = choose_a?
+      [p, 1-p]
+    end
+
+    class Variant
+      def initialize(args={})
+        @successes = args[:successes]
+        @failures = args[:failures]
+        @size = args[:size]
+        raise ArgumentError, "Invalid ABTest variant, you must pass 2 or more of :successes, :failures, :size and successes + failures = size" unless validate
+      end
+
+      def successes
+        @successes ||= calculate_successes
+      end
+
+      def failures
+        @failures ||= calculate_failures
+      end
+
+      def size
+        @size ||= calculate_size
+      end
+
+      def success_rate
+        successes/size.to_f
+      end
+
+      def failure_rate
+        failures/size.to_f
+      end
+
+      def ==(other)
+        size == other.size && successes == other.successes && failures == other.failures
+      end
+
+      private
+      def has_successes?
+        !!successes
+      end
+
+      def has_failures?
+        !!failures
+      end
+
+      def has_size?
+        !!size
+      end
+
+      def calculate_successes
+        @size - @failures if @size && @failures
+      end
+
+      def calculate_failures
+        @size - @successes if @size && @successes
+      end
+
+      def calculate_size
+        @successes + @failures if @successes && @failures
+      end
+
+      def validate
+        has_successes? && has_failures? && has_size? && (successes + failures == size)
+      end
+    end
+  end
+end

data/lib/abba/bayes.rb

@@ -0,0 +1,38 @@
+module ABBA
+  module Bayes
+    class BinaryOutcomeTest
+      def initialize(variant_a, variant_b)
+        @variant_a = variant_a
+        @variant_b = variant_b
+      end
+
+      def call
+        total = 0.0
+
+        (0..(alpha_a-1)).each do |i|
+          log_contribution = Math.lbeta(alpha_b+i, beta_a+beta_b) - Math.log(beta_a+i) - Math.lbeta(1+i, beta_a) - Math.lbeta(alpha_b, beta_b)
+          total += Math.exp(log_contribution)
+        end
+
+        total
+      end
+
+      private
+      def alpha_a
+        @variant_a.successes + 1
+      end
+
+      def beta_a
+        @variant_a.failures + 1
+      end
+
+      def alpha_b
+        @variant_b.successes + 1
+      end
+
+      def beta_b
+        @variant_b.failures + 1
+      end
+    end
+  end
+end

data/lib/abba/math.rb

@@ -0,0 +1,40 @@
+require 'singleton'
+
+module ABBA
+  module Math
+
+    class << self
+      def exp(x)
+        ::Math.exp(x)
+      end
+
+      def log(x)
+        ::Math.log(x)
+      end
+
+      def gamma(x)
+        ::Math.gamma(x)
+      end
+
+      def lgamma(x)
+        ::Math.lgamma(x)
+      end
+
+      def beta(x,y)
+        (gamma(x)*gamma(y))/(gamma(x+y))
+      end
+
+      def lbeta(x,y)
+        lgamma_x, sign_gamma_x = lgamma(x)
+        lgamma_y, sign_gamma_y = lgamma(y)
+        lgamma_xy, sign_gamma_xy = lgamma(x+y)
+
+        if (sign_gamma_x * sign_gamma_y * sign_gamma_xy) == 1
+          lgamma_x + lgamma_y - lgamma_xy
+        else
+          raise ::Math::DomainError, 'Numerical argument is out of domain - "lbeta"'
+        end
+      end
+    end
+  end
+end

data/lib/abba/version.rb

@@ -0,0 +1,3 @@
+module ABBA
+  VERSION = '0.1.0'
+end

metadata

@@ -0,0 +1,76 @@
+--- !ruby/object:Gem::Specification
+name: abba
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Kyle Tate
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-12-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.8'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: 
+email: kbt.tate@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/abba.rb
+- lib/abba/ab_test.rb
+- lib/abba/bayes.rb
+- lib/abba/math.rb
+- lib/abba/version.rb
+homepage: http://github.com/infiton/abba
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Bayesian A/B Testing
+test_files: []