darkhelmet-darkext 0.1.0 → 0.2.0

data/Manifest.txt

@@ -1,4 +1,3 @@
-History.txt
 Manifest.txt
 PostInstall.txt
 README.rdoc

data/README.rdoc

@@ -27,7 +27,7 @@ Just some useful Ruby functionality. No particular focus, except usefulness
 
 (The MIT License)
 
-Copyright (c) 2008 FIXME full name
+Copyright (c) 2008 Daniel Huckstep
 
 Permission is hereby granted, free of charge, to any person obtaining
 a copy of this software and associated documentation files (the

data/lib/darkext/io.rb

@@ -1,6 +1,6 @@
 require 'darkext/hash'
 
-module IO
+module DarkIO
   # Runs a block and captures the output it generates
   def self.capture_output(opts = { }) # yield e
     opts.with_defaults!(:stdout => true, :stderr => true)

data/lib/darkext/statistics.rb

@@ -1,12 +1,16 @@
 require 'mathn'
-require 'darkext/numeric'
+
 require 'darkext/array'
+require 'darkext/numeric'
+require 'darkext/symbol'
 
 class Array
   # Finds the mean of the array
   def mean
     self.sum / self.size.to_f
   end
+  alias :average :mean
+  alias :ave :mean
 
   # Finds the median of the array
   def median
@@ -99,4 +103,39 @@ module Statistics
     end
     return 1 - (probs[1] - probs[0])
   end
+
+  module Statistics::Regression
+    # Do a least squares linear regression on the two sets of x's and y's
+    # Returns a hash containing many relevant values
+    # * n (:n)
+    # * B_1 (:b_1)
+    # * B_0 (:b_0)
+    # * predicted values (:predicted)
+    # * residuals (:residuals)
+    # * SS_E (:ss_e)
+    # * unbiased estimator (:estimator)
+    # * the equation as a lambda (:equation)
+    # Raises an argument error if the arguments are not the same size
+    def self.least_squares(xs,ys)
+      raise ArgumentError("Arguments must be of equal size") if xs.size != ys.size
+      n = xs.size
+      b_1 = (xs.zip(ys).map(&:*).sum - ((ys.sum * xs.sum)/n))/(xs.map(&:square).sum - (xs.sum.square/n))
+      b_0 = ys.mean - b_1 * xs.mean
+      equation = lambda { |x| b_0 + b_1 * x }
+      predicted = xs.map(&lambda)
+      residuals = ys.zip(predicted).map(&:-)
+      ss_e = residuals.map(&:square).sum
+      estimator = ss_e/(n - 2)
+      reg = Hash.new
+      reg[:n] = n
+      reg[:b_1] = b_1
+      reg[:b_0] = b_0
+      reg[:predicted] = predicted
+      reg[:residuals] = residuals
+      reg[:ss_e] = ss_e
+      reg[:estimator] = estimator
+      reg[:equation] = equation
+      return reg
+    end
+  end
 end

metadata

@@ -1,15 +1,15 @@
 --- !ruby/object:Gem::Specification 
 name: darkhelmet-darkext
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Daniel Huckstep
 autorequire: 
 bindir: bin
-cert_chain: []
-
-date: 2008-11-23 00:00:00 -08:00
+cert_chain: 
+- /home/helmet/.gem/gem-public_cert.pem
+date: 2008-11-24 00:00:00 -08:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -38,12 +38,10 @@ executables: []
 extensions: []
 
 extra_rdoc_files: 
-- History.txt
 - Manifest.txt
 - PostInstall.txt
 - README.rdoc
 files: 
-- History.txt
 - Manifest.txt
 - PostInstall.txt
 - README.rdoc

data/History.txt

@@ -1,4 +0,0 @@
-== 0.0.1 2008-11-21
-
-* 1 major enhancement:
-  * Initial release