ruby-band 0.1.11

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

@@ -0,0 +1,190 @@
+$:.unshift File.dirname(__FILE__)
+require 'rules_utils'
+
+module Weka
+  module Classifier
+    module Rules
+      java_import "weka.classifiers.rules.DecisionTable" 
+      java_import "weka.classifiers.rules.DTNB"
+      java_import "weka.classifiers.rules.JRip"
+      java_import "weka.classifiers.rules.M5Rules"
+      java_import "weka.classifiers.rules.NNge"
+      java_import "weka.classifiers.rules.OneR"
+      java_import "weka.classifiers.rules.PART"
+      java_import "weka.classifiers.rules.Prism"
+      java_import "weka.classifiers.rules.Ridor"
+      java_import "weka.classifiers.rules.Rule"
+      java_import "weka.classifiers.rules.RuleStats"
+      java_import "weka.classifiers.rules.ZeroR"
+
+      class DecisionTable
+        include Rules_utils
+        class Base < DecisionTable
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class DTNB
+        include Rules_utils
+        class Base < DTNB
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class JRip
+        include Rules_utils
+        class Base < JRip
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class M5Rules
+        include Rules_utils
+        class Base < M5Rules
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class NNge
+        include Rules_utils
+        class Base < NNge
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class OneR
+        include Rules_utils
+        class Base < OneR
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class PART
+        include Rules_utils
+        class Base < PART
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class Prism
+        include Rules_utils
+        class Base < Prism
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class Ridor
+        include Rules_utils
+        class Base < Ridor
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class Rule
+        include Rules_utils
+        class Base < Rule
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class RuleStats
+        include Rules_utils
+        class Base < RuleStats
+          def initialize(&block)
+            super
+            if block_given?
+              init_instance_classifier(&block)
+            else
+              init_classifier  
+            end
+          end
+        end
+      end
+
+      class ZeroR
+        include Rules_utils
+        class Base < ZeroR
+          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/rules/rules_utils.rb

@@ -0,0 +1,81 @@
+#This module is used by the 'rules' classifiers from 'rules.rb' 
+#to inherit the following methods (instance and class methods)
+module Rules_utils
+  java_import "weka.core.Utils"
+
+  def init_rules
+    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 classifier
+  #ARGS:
+  #options -> a String, i.e. "-K 3"
+  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
+
+  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/trees/trees.rb

@@ -0,0 +1,110 @@
+$:.unshift File.dirname(__FILE__)
+require 'trees_utils'
+
+module Weka
+	module Classifier
+    #This module contains classifiers from the 'weka.classifiers.trees' package
+		module Trees
+      java_import 'weka.classifiers.trees.J48'
+      java_import 'weka.classifiers.trees.FT'
+      java_import 'weka.classifiers.trees.RandomForest'
+      java_import 'weka.classifiers.trees.BFTree'
+      java_import 'weka.classifiers.trees.J48graft'
+      java_import 'weka.classifiers.trees.LADTree'
+      java_import 'weka.classifiers.trees.LMT'
+      java_import 'weka.classifiers.trees.M5P'
+      java_import 'weka.classifiers.trees.RandomTree'
+
+      class ADTree
+        include Trees_utils
+        class Base < ADTree
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class BFTree
+        include Trees_utils
+        class Base < BFTree
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class FT
+        include Trees_utils
+        class Base < FT
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class J48
+        include Trees_utils
+        class Base < J48
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class J48graft
+        include Trees_utils
+        class Base < J48graft
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class LMT
+        include Trees_utils
+        class Base < LMT
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class M5P
+        include Trees_utils
+        class Base < M5P
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class RandomForest
+        include Trees_utils
+        class Base < RandomForest
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+      class RandomTree
+        include Trees_utils
+        class Base < RandomTree
+          def initialize
+            super
+            init_tree
+          end
+        end
+      end
+
+		end
+  end
+end

data/lib/ruby-band/weka/classifiers/trees/trees_utils.rb

@@ -0,0 +1,85 @@
+# This module is used by the 'trees' classifiers from 'trees.rb' 
+# to inherit the following methods (instance and class methods)
+module Trees_utils
+  java_import "weka.core.Utils"
+
+  def init_tree
+    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 the class index for the input dataset
+  def set_class_index(class_index)
+    @class_index = class_index 
+  end
+
+  #Set options for the instance classifier
+  #ARGS:
+  #options -> A String object, i.e. "-K 3"
+  def set_options(options)
+    options_inst = Utils.splitOptions(options)
+    setOptions(options_inst)
+  end
+
+  def list_options
+    listOptions.each {|key| puts "#{key.synopsis} #{key.description}"}
+  end
+
+  #Return a short description 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
+
+
+