Tree.hpp

Go to the documentation of this file.

#pragma once
 
#include "utils/Types.hpp"
 
#include "models/Bagged.hpp"
#include "models/Model.hpp"
#include "models/TreeNode.hpp"
 
 
namespace ppforest2 {
  class Tree : public Model {
  public:
    using Ptr = std::unique_ptr<Tree>;
    using Model::predict;
 
    using FeatureMatrix = types::FeatureMatrix;
    using FeatureVector = types::FeatureVector;
    using OutcomeVector = types::OutcomeVector;
    using Outcome       = types::Outcome;
    using RNG           = stats::RNG;
 
    using Root = TreeNode::Ptr;
 
    using GroupPartition = stats::GroupPartition;

    Root root;

    static Ptr train(TrainingSpec const& spec, types::FeatureMatrix& x, types::OutcomeVector& y);

    static Ptr train(TrainingSpec const& spec, types::FeatureMatrix& x, types::OutcomeVector& y, stats::RNG& rng);
 

    types::Outcome predict(types::FeatureVector const& x) const override;

    void accept(Model::Visitor& visitor) const override = 0;
 
    bool operator==(Tree const& other) const;
    bool operator!=(Tree const& other) const;
 
  protected:
    Tree(TreeNode::Ptr root, TrainingSpec::Ptr spec)
        : root(std::move(root)) {
      training_spec = std::move(spec);
      degenerate    = this->root && this->root->degenerate;
    }

    static Root
    build_root(TrainingSpec const& spec, FeatureMatrix& x, OutcomeVector& y, GroupPartition const& y_part, RNG& rng);
  };

  using BaggedTree = Bagged<Tree>;
}