TreeNode.hpp

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <set>
 
#include "utils/Types.hpp"
 
namespace ppforest2 {
  class TreeBranch;
  class TreeLeaf;

  class TreeNode {
  public:
    using Ptr = std::unique_ptr<TreeNode>;

    class Visitor {
    public:
      virtual ~Visitor() = default;
 
      virtual void visit(TreeBranch const&) {}
      virtual void visit(TreeLeaf const&) {}
    };

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

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

    virtual types::Outcome predict(types::FeatureVector const& x) const = 0;

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

    virtual int group_count() const = 0;

    virtual std::set<types::GroupId> node_groups() 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;
  };

  bool is_leaf(TreeNode const& node);
  bool is_leaf(TreeNode::Ptr const& node);
}