ruby-band 0.1.11

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

@@ -0,0 +1,78 @@
+#This module is used by the 'functions' classifiers from 'functions.rb' 
+#to inherit the following methods (instance and class methods)
+module Functions_utils
+  java_import "weka.core.Utils"
+
+  def init_function
+    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
+
+  #Instance methods list
+  def self.included(base)
+    base.extend(ClassMethods)
+  end
+
+  def init_instance_classifier(&block)
+    self.instance_eval(&block)
+    @dataset.setClassIndex(@class_index)
+    build_classifier(@dataset)
+  end
+
+  #Set instance data for the instance classifier
+  def set_data(data)
+    @dataset = data 
+  end
+
+  #Set a class index for the input dataset
+  def set_class_index(class_index)
+    @class_index = class_index 
+  end
+
+  def set_options(options)
+    options_inst = Utils.splitOptions(options)
+    setOptions(options_inst)
+  end
+
+  #List available options
+  def list_options
+    listOptions.map {|key| "#{key.synopsis} #{key.description}"}.join("\n")
+  end
+
+  #Return a description from the Weka Javadoc for the selected classifier
+  def description
+    puts 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/lazy/lazy.rb

@@ -0,0 +1,86 @@
+$:.unshift File.dirname(__FILE__)
+require 'lazy_utils'
+
+module Weka
+	module Classifier
+		module Lazy
+
+      java_import 'weka.classifiers.lazy.KStar' 
+      java_import 'weka.classifiers.lazy.LWL' 
+      java_import 'weka.classifiers.lazy.LBR'
+      java_import 'weka.classifiers.lazy.IB1'
+      java_import 'weka.classifiers.lazy.IBk'
+
+			class KStar
+				include Lazy_utils
+				class Base < KStar
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+				end
+			end
+
+      class LWL
+        include Lazy_utils
+        class Base < LWL
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class LBR
+        include Lazy_utils
+        class Base < LBR
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class IB1
+        include Lazy_utils
+        class Base < IB1
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class IBk
+        include Lazy_utils
+        class Base < IBk
+          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/lazy/lazy_utils.rb

@@ -0,0 +1,83 @@
+#This module is used by the 'lazy' classifiers from 'lazy.rb' 
+#to inherit the following methods (instance and class methods)
+module Lazy_utils
+  java_import "weka.core.Utils"
+
+  def init_lazy
+    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
+
+  #Instance methods list
+  def self.included(base)
+    base.extend(ClassMethods)
+  end
+
+  def init_instance_classifier(&block)
+    self.instance_eval(&block)
+    @dataset.setClassIndex(@class_index)
+    build_classifier(@dataset)
+  end
+
+  #Set data for instance classifier
+  #ARGV
+  # data -> an Instances object
+  def set_data(data)
+    @dataset = data 
+  end
+
+  #Set a class index for the input dataset
+  def set_class_index(class_index)
+    @class_index = class_index 
+  end
+
+  #Set options for the selected classfier
+  #ARGS:
+  #options -> a String, i.e. "-K"
+  def set_options(options)
+    options_inst = Utils.splitOptions(options)
+    setOptions(options_inst)
+  end
+
+  #List available options
+  def list_options
+    listOptions.map {|key| "#{key.synopsis} #{key.description}"}.join("\n")
+  end
+
+  #Return a description from the Weka JavaDoc for the selected classifier
+  def description
+    puts 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/mi/mi.rb

@@ -0,0 +1,191 @@
+$:.unshift File.dirname(__FILE__)
+require 'mi_utils'
+
+module Weka
+	module Classifier
+		module Mi
+      #This module contains classifiers from the 'weka.classifiers.mi' package
+      java_import 'weka.classifiers.mi.CitationKNN'
+      java_import 'weka.classifiers.mi.MDD'
+      java_import 'weka.classifiers.mi.MIBoost'
+      java_import 'weka.classifiers.mi.MIDD'
+      java_import 'weka.classifiers.mi.MIEMDD'
+      java_import 'weka.classifiers.mi.MILR'
+      java_import 'weka.classifiers.mi.MINND'
+      java_import 'weka.classifiers.mi.MIOptimalBall'
+      java_import 'weka.classifiers.mi.MISMO'
+      java_import 'weka.classifiers.mi.MISVM'
+      java_import 'weka.classifiers.mi.MIWrapper'
+      java_import 'weka.classifiers.mi.SimpleMI'
+
+      class CitationKNN
+        include Mi_utils
+        class Base < CitationKNN
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MIEMDD
+        include Mi_utils
+        class Base < MIEMDD
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MILR
+        include Mi_utils
+        class Base < MILR
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MINND
+        include Mi_utils
+        class Base < MINND
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MIOptimalBall
+        include Mi_utils
+        class Base < MIOptimalBall
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MISMO
+        include Mi_utils
+        class Base < MISMO
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MISVM
+        include Mi_utils
+        class Base < MISVM
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MIWrapper
+        include Mi_utils
+        class Base < MIWrapper
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MDD
+        include Mi_utils
+        class Base < MDD
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MIBoost
+        include Mi_utils
+        class Base < MIBoost
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class MIDD
+        include Mi_utils
+        class Base < MIDD
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class SimpleMI
+        include Mi_utils
+        class Base < SimpleMI
+          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/mi/mi_utils.rb

@@ -0,0 +1,80 @@
+#This module is used by the 'mi' classifiers from 'mi.rb' 
+#to inherit the following methods (instance and class methods)
+module Mi_utils
+  java_import "weka.core.Utils"
+
+  def init_mi
+    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
+
+  #Instance methods list
+  def self.included(base)
+    base.extend(ClassMethods)
+  end
+
+  def init_instance_classifier(&block)
+    self.instance_eval(&block)
+    @dataset.setClassIndex(@class_index)
+    build_classifier(@dataset)
+  end
+
+  #Set input data for the selected classifier
+  #ARGV:
+  #data -> an Instances class object
+  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
+
+  def set_options(options)
+    options_inst = Utils.splitOptions(options)
+    setOptions(options_inst)
+  end
+
+  #List options for the selected classifier
+  def list_options
+    listOptions.each {|key| puts "#{key.synopsis} #{key.description}"}
+  end
+
+  #Return a short description for the current classifier
+  def description
+    puts 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