darkhelmet-darkext 0.4.2 → 0.4.3

data/lib/darkext/statistics.rb

@@ -38,24 +38,22 @@ class Array
     map.keys.select { |x| map[x] == max }
   end
 
-  # Population variance
-  def pvariance
-    self.sum_of_squares.to_f / self.size.to_f
+  # Variance
+  def population_variance
+    self.sum_of_squares.to_f / (self.size).to_f
   end
 
-  # Sample variance
-  def svariance
+  def sample_variance
     self.sum_of_squares.to_f / (self.size - 1).to_f
   end
 
-  # Population standard deviation
-  def pdeviation
-    self.pvariance.abs.sqrt
+  # Standard deviation
+  def population_deviation
+    self.population_variance.abs.sqrt
   end
 
-  # Sample standard deviation
-  def sdeviation
-    self.svariance.abs.sqrt
+  def sample_deviation
+    self.sample_variance.abs.sqrt
   end
 
   # Randomly samples n elements
@@ -63,13 +61,24 @@ class Array
     (1..n).collect { self[rand(self.size)] }
   end
 
-  # Makes a two sided confidence interval for the array
-  # Percent must be 0 < percent < 1 for this to work properly
-  def ci(percent = 0.95, rho = 1)
+  # Generates a confidence interval
+  def ci(opts = { })
+    opts.with_defaults!({ :percent => 0.95, :rho => 1, :type => :center })
     m = self.mean
-    i = ((Darkext::Statistics::zscore((1 - percent) / 2) * rho) /
+    ret = Array.new
+    div = (opts[:type] == :center ? 2 : 1)
+    i = ((Darkext::Statistics::zscore((1 - percent) / div) * rho) /
          self.size.sqrt).abs
-    [m - i, m + i]
+    case opts[:type]
+    when :center
+      ret << m - i
+      ret << m + i
+    when :upper
+      ret << m + i
+    when :lower
+      ret << m - i
+    end
+    return ret
   end
 
   # Standardizes the array
@@ -80,17 +89,27 @@ class Array
   # Destructive standardize
   def standardize!
     m = self.mean.to_f
-    rho = self.deviation.to_f
-    self.map! do |v|
-      (v.to_f - m) / rho
-    end
+    rho = self.sample_deviation.to_f
+    self.map! { |v| (v.to_f - m) / rho }
   end
 
   def sum_of_squares
     m = self.mean
     self.map { |v| v - m }.map(&:square).sum
   end
+
+  # Normalize the Array
+  def normalize
+    self.clone.normalize!
+  end
+
+  # Normalize the Array destructive
+  def normalize!
+    m = self.mean.to_f
+    self.map! { |v| v / m }
+  end
 end
+
 module Darkext
   module Darkext::Statistics
     # Finds the probability of a z-score
@@ -108,7 +127,7 @@ module Darkext
       prob = 0
       while (maxz - minz) > epsilon
         prob = prob(zscore)
-        if prob > p then maxz = zscore else minz = zscore end
+        prob > p ? maxz = zscore : minz = zscore
         zscore = (maxz + minz) * 0.5
       end
       return zscore

metadata

@@ -1,15 +1,15 @@
 --- !ruby/object:Gem::Specification 
 name: darkhelmet-darkext
 version: !ruby/object:Gem::Version 
-  version: 0.4.2
+  version: 0.4.3
 platform: ruby
 authors: 
 - Daniel Huckstep
 autorequire: 
 bindir: bin
-cert_chain: 
-- /home/helmet/.gem/gem-public_cert.pem
-date: 2008-12-07 00:00:00 -08:00
+cert_chain: []
+
+date: 2008-12-11 00:00:00 -08:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency