TreeResponse.hpp

Go to the documentation of this file.

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

    types::Response value;
 
    explicit TreeResponse(types::Response value);
 
    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 true; }
 
    int group_count() const override { return 1; }
 
    std::set<types::Response> node_groups() const override { return {value}; }
 
    bool equals(TreeNode const& other) const override;
    TreeNode::Ptr clone() const override;

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