Module: RandomForester

Defined in:
lib/random_forester.rb,
lib/random_forester/version.rb

Constant Summary collapse

VERSION = 
"0.1.0"

Class Attribute Summary collapse

Class Method Summary collapse

Class Attribute Details

.logger ⇒ Object 



15
16
17
18
19
 
# File 'lib/random_forester.rb', line 15

def logger
  @logger ||= Logger.new($stdout).tap do |log|
    log.progname = self.name
  end
end

Class Method Details

.get_model(pmml_file_name) ⇒ Object 



22
23
24
25
 
# File 'lib/random_forester.rb', line 22

def self.get_model(pmml_file_name)
  xml = get_xml(pmml_file_name)
  new_model(xml)
end

.get_model_type(xml) ⇒ Object 



42
43
44
 
# File 'lib/random_forester.rb', line 42

def self.get_model_type(xml)
  xml.xpath("PMML/MiningModel/@modelName").to_s
end

.get_xml(pmml_file_name) ⇒ Object 



36
37
38
39
40
 
# File 'lib/random_forester.rb', line 36

def self.get_xml(pmml_file_name)
  pmml_string = File.open(pmml_file_name, 'rb').read
  xml = Nokogiri::XML(pmml_string)
  xml.remove_namespaces!
end

.new_model(xml) ⇒ Object 



27
28
29
30
31
32
33
34
 
# File 'lib/random_forester.rb', line 27

def self.new_model(xml)
  case get_model_type(xml)
    when RANDOM_FOREST_MODEL
      RandomForest.new(xml)
    else
      raise MODEL_NOT_SUPPORTED_ERROR
    end
end