TreeLeaf.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/TreeNode.hpp"
 
namespace ppforest2 {
  class TreeLeaf : public TreeNode {
  public:
    using Ptr = std::unique_ptr<TreeLeaf>;

    types::Outcome value;
 
    explicit TreeLeaf(types::Outcome value);
 
    void accept(TreeNode::Visitor& visitor) const override;
    types::Outcome response() const override;

    types::Outcome predict(types::FeatureVector const& x) const override;
 
    int group_count() const override { return 1; }
 
    std::set<types::GroupId> node_groups() const override { return {static_cast<types::GroupId>(value)}; }
 
    bool equals(TreeNode const& other) const override;
    TreeNode::Ptr clone() const override;

    static Ptr make(types::Outcome value);
  };
}