TreeDefs.h

Go to the documentation of this file.

#ifndef TREE_DEFS_H
#define TREE_DEFS_H
 
#include "CNum/DataStructs/DataStructs.h"
#include <utility>
#include <variant>

namespace CNum::Model::Tree {
  using SplitValuePair = std::pair<double, double>;
  
  constexpr int N_BINS = 256;
 
  using DataMatrix = std::variant< CNum::DataStructs::Matrix<int>, CNum::DataStructs::Matrix<double> >;

  struct Histogram {
    arena_view_t g_bin;
    arena_view_t h_bin;
  };

  struct DataPartition {
    arena_view_t *global_idx_array;
    size_t start;
    size_t end;
  };
  
  struct Split {
    int feature;
    double threshold;
    double best_gain;
    int bin;
    SplitValuePair values;
  };

  enum split_dir {
    LEFT,
    RIGHT
  };
 
  class TreeBoosterNode;
  class TreeBooster;
  class XGTreeBooster;
  template <typename T> class GBModel;
};
 
#endif