RegressionTree.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/Tree.hpp"
 
namespace ppforest2 {
  class RegressionTree : public Tree {
  public:
    using Ptr = std::unique_ptr<RegressionTree>;
    using Tree::predict;
 
    using FeatureMatrix = types::FeatureMatrix;
    using FeatureVector = types::FeatureVector;
    using OutcomeVector = types::OutcomeVector;
    using Outcome       = types::Outcome;
    using RNG           = stats::RNG;
 
    using GroupPartition = stats::GroupPartition;
 
    RegressionTree(TreeNode::Ptr root, TrainingSpec::Ptr spec)
        : Tree(std::move(root), std::move(spec)) {
      invariant(this->training_spec != nullptr, "RegressionTree requires a non-null TrainingSpec");
      invariant(is_regression(this), "RegressionTree requires a regression TrainingSpec");
    }

    static Ptr train(TrainingSpec const& s, FeatureMatrix& x, OutcomeVector& y, GroupPartition const& y_part, RNG& rng);
 
    void accept(Model::Visitor& visitor) const override;
  };
}