kder 0.0.1

data/lib/kder.rb

@@ -0,0 +1,5 @@
+require_relative 'kder/kde'
+class Kder
+  Major,Minor,Patch = File.readlines("VERSION").map(&:chomp)
+  VERSION = [Major,Minor,Patch].join(".")
+end

data/lib/kder/bandwidth.rb

@@ -0,0 +1,13 @@
+require_relative '../util/enum_extensions'
+module Bandwidth
+  class << self
+    def silverman(input_arr, std_deviation = input_arr.standard_deviation) 
+      bandwidth = ((4 * std_deviation ** 5)/(3 * input_arr.size))**(1.0/5)
+    end
+    alias_method :estimate, :silverman 
+    alias_method :nrd0, :silverman
+
+    #def mean_integrated_squared_error(a,b,c,d)
+    #end
+  end
+end

data/lib/kder/kde.rb

@@ -0,0 +1,35 @@
+class Kder
+  require 'distribution'
+  require_relative 'bandwidth'
+  require_relative '../util/statistics'
+
+  Sigmas = 2.5 
+  MeshCount = 2e3
+  MinimumThresholdValue = 1e-2
+  class << self
+    ## 
+    # :singleton-method: kde
+    # Accepts a single member array plus optional additional information
+    # Returns a two member array, [x_vals,y_vals] representing the kde
+    def kde(arr, bw = nil, opts = {sigmas: Sigmas, sampling_density: MeshCount, threshold: MinimumThresholdValue})
+      unless bw # is nil
+        bw = Bandwidth.silverman(arr)
+      end
+      # Initialization steps
+      min = arr.min - bw*opts[:sigmas]
+      max = arr.max + bw*opts[:sigmas]
+      step_size = (max-min)/opts[:sampling_density].to_f
+      arr.sort!
+      # Step through the range
+      output = (min..max).step(step_size).map do |mid|
+        high_end = mid+ bw*opts[:sigmas]
+        lower_end = mid - bw*opts[:sigmas]
+        included = arr.select {|a| (lower_end..high_end).include?(a)}
+        intensity = included.map {|a| Kder::Statistics.custom_pdf(a-mid, bw) }.inject(:+)
+        intensity ||= 0
+        [mid, intensity ] unless intensity < opts[:threshold]
+      end
+      output.compact.transpose
+    end
+  end
+end

data/lib/util/enum_extensions.rb

@@ -0,0 +1,19 @@
+module Enumerable
+  # Thanks StackOverflow users!
+  # http://stackoverflow.com/questions/7749568/how-can-i-do-standard-deviation-in-ruby
+  def sum
+    self.inject(:+)
+  end
+  def mean
+    self.sum/self.length.to_f
+  end
+  alias average mean
+  def sample_variance
+    m = self.mean
+    sum = self.inject(0){|accum, i| accum + (i-m)**2}
+    sum/(self.length - 1).to_f
+  end
+  def standard_deviation
+    Math.sqrt(self.sample_variance)
+  end
+end

data/lib/util/statistics.rb

@@ -0,0 +1,10 @@
+class Kder
+  module Statistics
+    SQ2PI = Math.sqrt(2 * Math::PI)
+    # Normal pdf with a custom sigma value, still centered around 0.
+    def self.custom_pdf(x, sigma = 1.0)
+      mean = 0 
+      (1.0/( sigma*SQ2PI))*Math::exp(-0.5*((x-mean)/sigma)**2)
+    end
+  end
+end

metadata

@@ -0,0 +1,52 @@
+--- !ruby/object:Gem::Specification
+name: kder
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Ryan Taylor
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-24 00:00:00.000000000 Z
+dependencies: []
+description: A KDE implementation in Ruby.
+email:
+- ryanmt@byu.net
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/kder.rb
+- lib/kder/bandwidth.rb
+- lib/kder/kde.rb
+- lib/util/enum_extensions.rb
+- lib/util/statistics.rb
+homepage: https://github.com/princelab/KDER
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Kernel Density Estimation, and an associated bandwidth estimator, rudimentary
+  at best, but a decent first approximation for backend preparing of KDE plots for
+  plotting in JS libraries, or for whatever you would like to use it.
+test_files: []