class Scoruby::Models::GradientBoostedModel::Model

Public Class Methods

new(xml) click to toggle source
# File lib/scoruby/models/gradient_boosted_model/model.rb, line 15
def initialize(xml)
  @data = Data.new(xml)
end

Public Instance Methods

score(features) click to toggle source
# File lib/scoruby/models/gradient_boosted_model/model.rb, line 19
def score(features)
  formatted_features = Features.new(features).formatted
  scores = traverse_trees(formatted_features)
  sum = scores.reduce(:+) + const
  Math.exp(sum) / (1 + Math.exp(sum))
end
traverse_trees(formatted_features) click to toggle source
# File lib/scoruby/models/gradient_boosted_model/model.rb, line 26
def traverse_trees(formatted_features)
  decision_trees.map do |dt|
    dt.decide(formatted_features).score.to_s.to_f
  end
end