rubythinking 0.2.0

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/rubythinking/distributions/binomial.rb

@@ -0,0 +1,29 @@
+class Rubythinking::Distributions::Binomial
+  class << self
+    def samples(n, size:, success:)
+      dist = Croupier::Distributions::Binomial.new(size: size, success: success)
+      samples = dist.to_enum.take(n)
+    end
+
+    def density(value:, size:, success:)
+      # Use variable names often used in math
+      # textbooks and wikipedia to have code
+      # close to the formula
+      k = value
+      n = size
+      p = success
+
+      (factorial(n).to_f / (factorial(k) * factorial(n - k))).to_f * (p**k) * ((1-p)**(n-k))
+    end
+
+    def factorial(n)
+      return 1 if n < 1
+      n.to_i.downto(1).inject(:*)
+    end
+
+
+    def likelihood(w, l, p)
+      density(value: w, size: (w+l), success: p)
+    end
+  end
+end

data/lib/rubythinking/distributions.rb

@@ -0,0 +1,4 @@
+module Rubythinking
+  module Distributions
+  end
+end

data/lib/rubythinking/version.rb

@@ -0,0 +1,3 @@
+module Rubythinking
+  VERSION = "0.2.0"
+end

data/lib/rubythinking.rb

@@ -0,0 +1,17 @@
+require "rubythinking/version"
+require "rubythinking/distributions"
+require "rubythinking/distributions/binomial"
+require "vega"
+require "cmd_stan_rb"
+require "croupier"
+
+module Rubythinking
+  # Mimicks R API
+  def dbinom(value, size:, prob:)
+    Rubythinking::Distributions::Binomial.density(value: value, success: prob, size: size)
+  end
+
+  def rbinom(n, size:, prob:)
+    Rubythinking::Distributions::Binomial.samples(n, success: prob, size: size)
+  end
+end

data/local_reinstall.sh

@@ -0,0 +1 @@
+gem build rubythinking.gemspec && gem install rubythinking-0.1.0.gem && iruby register --force