TreeNode.hpp

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <set>
 
#include "utils/Types.hpp"
 
namespace ppforest2 {
  struct TreeCondition;
  struct TreeResponse;

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

    struct Visitor {
      virtual void visit(TreeCondition const& condition) = 0;
      virtual void visit(TreeResponse const& response)   = 0;
    };

    bool degenerate = false;
 
    virtual ~TreeNode() = default;

    virtual void accept(Visitor& visitor) const = 0;

    virtual types::Response predict(types::FeatureVector const& data) const = 0;

    virtual types::Response response() const = 0;

    virtual int group_count() const = 0;

    virtual std::set<types::Response> node_groups() const = 0;

    virtual bool is_leaf() const = 0;

    virtual bool equals(TreeNode const& other) const = 0;

    virtual Ptr clone() const = 0;
 
    bool operator==(TreeNode const& other) const;
    bool operator!=(TreeNode const& other) const;
  };
}