TreeCondition.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/TreeNode.hpp"
#include "models/Projector.hpp"
#include "utils/Types.hpp"
#include "utils/Math.hpp"
 
namespace ppforest2 {
  using Threshold = types::Feature;

  struct TreeCondition final : public TreeNode {
    using Ptr = std::unique_ptr<TreeCondition>;

    pp::Projector projector;
    Threshold threshold;
    TreeNode::Ptr lower;
    TreeNode::Ptr upper;

    std::set<types::Response> groups;
    types::Feature pp_index_value = 0;
 
    TreeCondition(pp::Projector projector,
                  Threshold threshold,
                  TreeNode::Ptr lower,
                  TreeNode::Ptr upper,
                  std::set<types::Response> groups = {},
                  types::Feature pp_index_value    = 0);
 
    void accept(TreeNode::Visitor& visitor) const override;

    types::Response response() const override;

    types::Response predict(types::FeatureVector const& data) const override;
 
    bool is_leaf() const override { return false; }
 
    int group_count() const override { return static_cast<int>(groups.size()); }
 
    std::set<types::Response> node_groups() const override { return groups; }
 
    bool equals(TreeNode const& other) const override;
    TreeNode::Ptr clone() const override;

    static Ptr make(pp::Projector projector,
                    Threshold threshold,
                    TreeNode::Ptr lower,
                    TreeNode::Ptr upper,
                    std::set<types::Response> groups = {},
                    types::Feature pp_index_value    = 0);
  };
}