NodeContext.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/Projector.hpp"
#include "models/strategies/vars/VariableSelection.hpp"
#include "stats/GroupPartition.hpp"
#include "utils/Types.hpp"
 
#include <optional>
 
using Projector = ppforest2::pp::Projector;
 
namespace ppforest2 {
  struct NodeContext {
    types::FeatureMatrix& x;
    stats::GroupPartition const& y;
    int depth;

    bool aborted = false;

    types::OutcomeVector& y_vec;

    std::optional<vars::VariableSelection::Result> var_selection;

    std::optional<Projector> projector;
    std::optional<types::Feature> pp_index_value;

    std::optional<stats::GroupPartition> y_bin;

    std::optional<types::Feature> cutpoint;

    std::optional<types::GroupId> lower_group;
    std::optional<types::GroupId> upper_group;

    std::optional<stats::GroupPartition> lower_y_part;
    std::optional<stats::GroupPartition> upper_y_part;
 
    NodeContext(types::FeatureMatrix& x, stats::GroupPartition const& y, types::OutcomeVector& y_vec, int depth)
        : x(x)
        , y(y)
        , depth(depth)
        , y_vec(y_vec) {}

    stats::GroupPartition const& active_partition() const { return y_bin.has_value() ? *y_bin : y; }
  };
}