arroz 0.0.4

data/lib/arroz.rb

@@ -0,0 +1,5 @@
+module Arroz
+  require 'arroz/knn'
+  require 'arroz/ensemble'
+  require 'arroz/distances/euclidean'
+end

data/lib/arroz/distances/euclidean.rb

@@ -0,0 +1,12 @@
+module Arroz::Distances
+  module Euclidean
+    extend self
+
+    def distance(e1, e2)
+      sum = e1.zip(e2).map { |pair| (pair.first.to_f - pair.last.to_f)**2 }.
+        reduce(:+)
+
+      Math.sqrt(sum)
+    end
+  end
+end

data/lib/arroz/ensemble.rb

@@ -0,0 +1,35 @@
+module Arroz
+
+  class Ensemble
+
+    attr_accessor :classifiers
+
+    def initialize(classifiers)
+      @classifiers = classifiers
+    end
+
+    def classify(element)
+      guesses = classifiers.map { |c| c.classify(element) }
+      
+      plurality_vote(guesses)
+    end
+
+    def classify_projections(projections)
+      raise ArgumentError.new("Wrong number of projections") if projections.size != classifiers.size
+      
+      guesses = classifiers.zip(projections).
+        map { |classifier, projection| classifier.classify(projection) }
+
+      plurality_vote(guesses)
+    end
+
+  private
+
+    def plurality_vote(guesses)
+      uniq_guesses = guesses.uniq
+
+      uniq_guesses.map { |g| guesses.count(g) }.zip(uniq_guesses).
+        sort.last.last
+    end
+  end
+end

data/lib/arroz/knn.rb

@@ -0,0 +1,39 @@
+module Arroz
+  class KNN
+    
+    attr_reader :data, :k
+
+    # The label attribute in the data array must be in the first position
+    def initialize(data, k=1)
+      @data = data
+      @k = k
+    end
+
+    ## Returns the predicted class of the element passed
+    def classify(element)
+      neighbors = nearest_neighbors(element)
+
+      neighbor_classes = neighbors.map(&:last)
+      classes = neighbor_classes.uniq
+
+      classes.zip(classes.map { |c| neighbor_classes.count(c) }).
+        sort_by(&:last).last.first
+    end
+
+  private 
+
+    def nearest_neighbors(element)
+      distances = []
+      
+      @data.each do |datum|
+        distances << [distance(element, datum[1..-1]), datum.first]
+      end
+
+      distances.sort.take(k)
+    end
+
+    def distance(e1, e2)
+      raise NotImplementedError
+    end
+  end
+end

data/lib/arroz/version.rb

@@ -0,0 +1,3 @@
+module Arroz
+  VERSION = '0.0.4'
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification
+name: arroz
+version: !ruby/object:Gem::Version
+  version: 0.0.4
+  prerelease: 
+platform: ruby
+authors:
+- Guilherme Carvalho
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Simple attempt to build an IA framework in Ruby
+email:
+- guilherme@guava.com.br
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/arroz/distances/euclidean.rb
+- lib/arroz/ensemble.rb
+- lib/arroz/knn.rb
+- lib/arroz/version.rb
+- lib/arroz.rb
+homepage: http://github.com/gvc/arroz
+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: 1.3.6
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Simple attempt to build an IA framework in Ruby
+test_files: []