ModelUtils.h

Go to the documentation of this file.

#ifndef MODEL_UTILS_H
#define MODEL_UTILS_H
 
#include "CNum/DataStructs/DataStructs.h"
#include "CNum/Model/Tree/Tree.h"
#include "CNum/Utils/Utils.h"

namespace CNum::Utils::ModelUtils {
  ::std::unique_ptr<CNum::DataStructs::Matrix<double>[] > train_test_split(const CNum::DataStructs::Matrix<double> &X,
                                       const CNum::DataStructs::Matrix<double> &y,
                                       double test_percentage = 0.2,
                                       bool shuffle = true,
                                       uint64_t logical_id = 0);

  void print_tree(CNum::Model::Tree::TreeBoosterNode *node, int spaces = 10);
};
 
#endif