Module: Trees_utils

Included in:
Weka::Classifier::Trees::ADTree, Weka::Classifier::Trees::BFTree, Weka::Classifier::Trees::FT, Weka::Classifier::Trees::J48, Weka::Classifier::Trees::J48graft, Weka::Classifier::Trees::LMT, Weka::Classifier::Trees::M5P, Weka::Classifier::Trees::RandomForest, Weka::Classifier::Trees::RandomTree
Defined in:
lib/ruby-band/weka/classifiers/trees/trees_utils.rb

Overview

This module is used by the ‘trees’ classifiers from ‘trees.rb’ to inherit the following methods (instance and class methods)

Defined Under Namespace

Modules: ClassMethods

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(base) ⇒ Object

Instance methods list



13
14
15
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 13

def self.included(base)
  base.extend(ClassMethods)
end

Instance Method Details

#cross_validate(fold) ⇒ Object

perform crossvalidation on a trained classifier ARGV: fold -> ‘int’ value



55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 55

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

#description ⇒ Object

Return a short description for the selected classifier



48
49
50
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 48

def description
  puts globalInfo
end

#init_instance_classifier(&block) ⇒ Object 



17
18
19
20
21
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 17

def init_instance_classifier(&block)
  self.instance_eval(&block)
  @dataset.setClassIndex(@class_index)
  build_classifier(@dataset)
end

#init_tree ⇒ Object 



6
7
8
9
10
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 6

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

#list_options ⇒ Object 



43
44
45
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 43

def list_options
  listOptions.each {|key| puts "#{key.synopsis} #{key.description}"}
end

#set_class_index(class_index) ⇒ Object

Set the class index for the input dataset



31
32
33
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 31

def set_class_index(class_index)
  @class_index = class_index 
end

#set_data(data) ⇒ Object

Set input data for the selected classifier ARGV: data -> an Instances class object



26
27
28
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 26

def set_data(data)
  @dataset = data 
end

#set_options(options) ⇒ Object

Set options for the instance classifier ARGS: options -> A String object, i.e. “-K 3”



38
39
40
41
 
# File 'lib/ruby-band/weka/classifiers/trees/trees_utils.rb', line 38

def set_options(options)
  options_inst = Utils.splitOptions(options)
  setOptions(options_inst)
end