Train.hpp

Go to the documentation of this file.

#pragma once
 
#include "cli/CLIOptions.hpp"
#include "io/Output.hpp"
#include "stats/DataPacket.hpp"
#include "stats/Stats.hpp"
#include "utils/Types.hpp"
#include "models/Model.hpp"
 
namespace CLI {
  class App;
}
 
namespace ppforest2::cli {
  CLI::App* setup_train(CLI::App& app, CLIOptions& params);

  void add_model_options(CLI::App* sub, ModelParams& model);

  struct TrainResult {
    ppforest2::Model::Ptr model;
    long long duration;
  };

  ppforest2::stats::DataPacket read_data(CLIOptions const& params, ppforest2::stats::RNG& rng);

  TrainResult train_model(ppforest2::types::FeatureMatrix const& x,
                          ppforest2::types::ResponseVector const& y,
                          CLIOptions const& params,
                          ppforest2::stats::RNG& rng);

  int run_train(CLIOptions& params);
}