Class: Mjai::Manue::DangerEstimator::DecisionTree
- Inherits:
-
Object
- Object
- Mjai::Manue::DangerEstimator::DecisionTree
- Defined in:
- lib/mjai/manue/danger_estimator.rb
Instance Method Summary collapse
- #estimate_prob(scene, pai) ⇒ Object
-
#initialize(path) ⇒ DecisionTree
constructor
A new instance of DecisionTree.
Constructor Details
#initialize(path) ⇒ DecisionTree
Returns a new instance of DecisionTree.
424 425 426 |
# File 'lib/mjai/manue/danger_estimator.rb', line 424 def initialize(path) @root = open(path, "rb"){ |f| Marshal.load(f) } end |
Instance Method Details
#estimate_prob(scene, pai) ⇒ Object
428 429 430 431 432 433 434 435 436 437 438 439 440 |
# File 'lib/mjai/manue/danger_estimator.rb', line 428 def estimate_prob(scene, pai) feature_vector = scene.feature_vector(pai.remove_red()) p [pai, DangerEstimator.feature_vector_to_str(feature_vector)] node = @root while node.feature_name if DangerEstimator.get_feature_value(feature_vector, node.feature_name) node = node.positive else node = node.negative end end return node.average_prob end |