ruby-band 0.1.11

data/lib/ruby-band/core/type/utils.rb

@@ -0,0 +1,31 @@
+require 'java'
+
+module Core
+ 	module Utils
+
+		# Convert a bidimensional RubyArray to a java double [][]
+		def Utils.bidimensional_to_double(ruby_array)
+			ruby_array.to_java Java::double[]
+		end
+
+    def Utils.bidimensional_to_long(ruby_array)
+      ruby_array.to_java Java::long[]
+    end
+
+		def Utils.double_to_a(java_array)
+			ruby_array = []
+			java_array.each {|val| ruby_array << val}
+			ruby_array
+		end
+
+  end
+end
+
+class Array
+  def is_2d?
+    self.each do |item|
+      return true if item.is_a? Array 
+    end
+    return false
+  end
+end

data/lib/ruby-band/weka.rb

@@ -0,0 +1,14 @@
+require 'ruby-band/weka/db/db'
+require 'ruby-band/weka/filters/unsupervised/attribute/attribute'
+require 'ruby-band/weka/filters/unsupervised/instance/instance'
+require 'ruby-band/weka/filters/supervised/instance/instance'
+require 'ruby-band/weka/filters/supervised/attribute/attribute'
+require 'ruby-band/weka/classifiers/bayes/bayes'
+require 'ruby-band/weka/classifiers/evaluation'
+require 'ruby-band/weka/classifiers/functions/functions'
+require 'ruby-band/weka/classifiers/trees/trees'
+require 'ruby-band/weka/classifiers/lazy/lazy'
+require 'ruby-band/weka/classifiers/rules/rules'
+require 'ruby-band/weka/clusterers/clusterers'
+require 'ruby-band/weka/attribute_selection/evaluators'
+require 'ruby-band/weka/attribute_selection/search'

data/lib/ruby-band/weka/attribute_selection/attribute_selection_utils.rb

@@ -0,0 +1,20 @@
+module Attribute_selection_Utils
+  java_import "weka.core.Utils"
+
+  #Instance methods list
+  def options_list
+    listOptions.map {|key| "#{key.synopsis} #{key.description}"}.join("\n")
+  end 
+
+  #Set options for an evaluator or a search algorithm
+  def select_options(options_string)
+    options = Utils.splitOptions(options_string)
+    set_options(options)
+  end
+
+  #Return a short description for the selected evalutator object or search algorithm
+  def description
+    globalInfo
+ end
+
+end

data/lib/ruby-band/weka/attribute_selection/evaluators.rb

@@ -0,0 +1,58 @@
+$:.unshift File.dirname(__FILE__)
+require 'attribute_selection_utils'
+
+module Weka
+  module Attribute_selection
+    module Evaluator
+      #This module contains evaluators from the 'weka.attributeSelection' packages
+      java_import 'weka.attributeSelection.CfsSubsetEval'
+      java_import 'weka.attributeSelection.ChiSquaredAttributeEval'
+      java_import 'weka.attributeSelection.ClassifierSubsetEval'
+      java_import 'weka.attributeSelection.ConsistencySubsetEval'
+      java_import 'weka.attributeSelection.CostSensitiveAttributeEval'
+      java_import 'weka.attributeSelection.CostSensitiveSubsetEval'
+      java_import 'weka.attributeSelection.FilteredAttributeEval'
+      java_import 'weka.attributeSelection.FilteredSubsetEval'
+      java_import 'weka.attributeSelection.GainRatioAttributeEval'
+      java_import 'weka.attributeSelection.SVMAttributeEval'
+
+      class CfsSubsetEval
+        include Attribute_selection_Utils 
+      #  java_alias :use_options , :setOptions, [Java::Java.lang.String[]]
+      end
+
+      class ChiSquaredAttributeEval
+        include Attribute_selection_Utils  
+      end
+
+      class ClassifierSubsetEval
+        include Attribute_selection_Utils  
+      end
+
+      class ConsistencySubsetEval
+        include Attribute_selection_Utils  
+      end
+
+      class CostSensitiveSubsetEval
+        include Attribute_selection_Utils  
+      end
+
+      class FilteredAttributeEval
+        include Attribute_selection_Utils  
+      end      
+
+      class FilteredSubsetEval
+        include Attribute_selection_Utils  
+      end
+
+      class GainRatioAttributeEval
+        include Attribute_selection_Utils  
+      end
+
+      class SVMAttributeEval
+        include Attribute_selection_Utils  
+      end
+
+    end 
+  end
+end

data/lib/ruby-band/weka/attribute_selection/search.rb

@@ -0,0 +1,52 @@
+$:.unshift File.dirname(__FILE__)
+require 'attribute_selection_utils'
+
+module Weka
+  module Attribute_selection
+    #This module contains search algorithms from the 'weka.attributeSelection' packages
+    module Search
+      
+      java_import 'weka.attributeSelection.GreedyStepwise'
+      java_import 'weka.attributeSelection.RankSearch'
+      java_import 'weka.attributeSelection.Ranker'
+      java_import 'weka.attributeSelection.ExhaustiveSearch'
+      java_import 'weka.attributeSelection.GeneticSearch'
+      java_import 'weka.attributeSelection.RaceSearch'
+      java_import 'weka.attributeSelection.RandomSearch'
+      java_import 'weka.attributeSelection.ScatterSearchV1'
+
+      class GreedyStepwise
+        include Attribute_selection_Utils 
+      end
+
+      class Ranker
+        include Attribute_selection_Utils 
+      end
+
+      class RankSearch
+        include Attribute_selection_Utils 
+      end
+
+      class ExhaustiveSearch
+        include Attribute_selection_Utils 
+      end
+
+      class GeneticSearch
+        include Attribute_selection_Utils 
+      end
+
+      class RaceSearch
+        include Attribute_selection_Utils 
+      end
+
+      class RandomSearch
+        include Attribute_selection_Utils 
+      end
+
+      class ScatterSearchV1
+        include Attribute_selection_Utils 
+      end
+
+    end 
+  end
+end

data/lib/ruby-band/weka/classifiers/bayes/bayes.rb

@@ -0,0 +1,86 @@
+$:.unshift File.dirname(__FILE__)
+require 'bayes_utils'
+
+module Weka
+  module Classifier
+    #This module stores the classifiers from the 'weka.classifiers.bayes' package
+    module Bayes
+      java_import "weka.classifiers.bayes.NaiveBayes"
+      java_import "weka.classifiers.bayes.BayesianLogisticRegression"
+      java_import "weka.classifiers.bayes.AODE"
+      java_import "weka.classifiers.bayes.ComplementNaiveBayes"
+      java_import "weka.classifiers.bayes.WAODE"
+
+      class NaiveBayes
+        include Bayes_utils
+        class Base < NaiveBayes
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class AODE
+        include Bayes_utils
+        class Base < AODE
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class BayesianLogisticRegression
+        include Bayes_utils
+        class Base < BayesianLogisticRegression
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class ComplementNaiveBayes
+        include Bayes_utils
+        class Base < ComplementNaiveBayes
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class WAODE
+        include Bayes_utils
+        class Base < WAODE
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/ruby-band/weka/classifiers/bayes/bayes_utils.rb

@@ -0,0 +1,82 @@
+#This module is used by the Bayesian classifiers from 'bayes.rb' 
+#to inherit the following methods (instance and class methods)
+module Bayes_utils
+
+  java_import "weka.core.Utils"
+
+  def init_classifier
+    set_options(self.class.options) if self.class.options 
+    self.class.data.setClassIndex(self.class.class_index) if self.class.class_index
+    buildClassifier(self.class.data)
+  end
+
+  def init_instance_classifier(&block)
+    self.instance_eval(&block)
+    @dataset.setClassIndex(@class_index)
+    build_classifier(@dataset)
+  end
+
+  # set data for instance classifier
+  def set_data(data)
+    @dataset = data 
+  end
+
+  # set class index for the input dataset
+  def set_class_index(class_index)
+    @class_index = class_index 
+  end
+
+  #Instance methods list
+  def self.included(base)
+    base.extend(ClassMethods)
+  end
+
+  # set classifier options
+  def set_options(options)
+    options_inst = Utils.splitOptions(options)
+    setOptions(options_inst)
+  end
+
+  def list_options
+    listOptions.map {|key| "#{key.synopsis} #{key.description}"}.join("\n")
+  end
+
+  # return the description reported in the Weka Java doc
+  def description
+    globalInfo
+  end
+
+  # perform crossvalidation on a trained classifier
+  #ARGV:
+  #fold -> 'int' value
+  def cross_validate(fold)
+    if self.class.data
+      eval = Weka::Classifier::Evaluation.new self.class.data
+      eval.crossValidateModel(self.class.ancestors[2].new, self.class.data, fold.to_java(:int), Random.new(1))
+      eval.summary 
+    else
+      eval = Weka::Classifier::Evaluation.new @dataset
+      eval.crossValidateModel(self.class.ancestors[1].new, @dataset, fold.to_java(:int), Random.new(1))
+      eval.summary
+    end
+  end
+
+  #Class methods module
+  module ClassMethods
+    
+    def self.classifier_attr_accessor(*args)
+      args.each do |arg|
+        #Here's the getter
+        self.class_eval("def #{arg};@#{arg};end")
+        #Here's the setter
+        self.class_eval("def set_#{arg}(val);@#{arg}=val;end")
+      end
+    end
+
+    classifier_attr_accessor :options,:data,:class_index
+
+  end
+end
+
+
+

data/lib/ruby-band/weka/classifiers/evaluation.rb

@@ -0,0 +1,13 @@
+module Weka
+  module Classifier
+    java_import 'weka.classifiers.Evaluation'
+
+    # Weka Evaluation class to be used with classfiers and clusterers
+    class Evaluation
+      def summary
+        toSummaryString
+      end
+    end
+    
+  end
+end

data/lib/ruby-band/weka/classifiers/functions/functions.rb

@@ -0,0 +1,177 @@
+$:.unshift File.dirname(__FILE__)
+require 'functions_utils'
+
+module Weka
+	module Classifier
+    #This module stores the classifiers from the 'weka.classifiers.functions' package
+		module Functions
+      java_import 'weka.classifiers.functions.LinearRegression'
+      java_import 'weka.classifiers.functions.PLSClassifier'
+      java_import 'weka.classifiers.functions.RBFNetwork'
+      java_import 'weka.classifiers.functions.GaussianProcesses'
+      java_import 'weka.classifiers.functions.LeastMedSq'
+      java_import 'weka.classifiers.functions.LibSVM'
+      java_import 'weka.classifiers.functions.SMO'
+      java_import 'weka.classifiers.functions.SMOreg'
+      java_import 'weka.classifiers.functions.SPegasos'
+      java_import 'weka.classifiers.functions.VotedPerceptron'
+      java_import 'weka.classifiers.functions.Winnow'
+
+
+      class LeastMedSq
+        include Functions_utils
+        class Base < LeastMedSq
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class LibSVM
+        include Functions_utils
+        class Base < LibSVM
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class LinearRegression
+        include Functions_utils
+        class Base < LinearRegression
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class PLSClassifier
+        include Functions_utils
+        class Base < PLSClassifier
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class RBFNetwork
+        include Functions_utils
+        class Base < RBFNetwork
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class SMO
+        include Functions_utils
+        class Base < SMO
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class SMOreg
+        include Functions_utils
+        class Base < SMOreg
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class SPegasos
+        include Functions_utils
+        class Base < SPegasos
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end 
+
+      class GaussianProcesses
+        include Functions_utils
+        class Base < GaussianProcesses
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class VotedPerceptron
+        include Functions_utils
+        class Base < VotedPerceptron
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class Winnow
+        include Functions_utils
+        class Base < Winnow
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+    end
+  end
+end