main/cpp/Train_8hpp_source.html

#pragma once


#include "cli/CLIOptions.hpp"

#include "utils/Types.hpp"

#include "models/Model.hpp"


namespace CLI {

  class App;

}


namespace ppforest2::cli {

  void setup_train(CLI::App& app, Params& params);


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


  struct TrainResult {

    ppforest2::Model::Ptr model;

    long long duration;

  };


  TrainResult

  train_model(ppforest2::types::FeatureMatrix const& x, ppforest2::types::OutcomeVector const& y, Params const& params);


  int run_train(Params& params);

}

CLIOptions.hpp
CLI argument parsing, validation, and configuration for ppforest2.

Model.hpp

Types.hpp

ppforest2::Model::Ptr
std::shared_ptr< Model > Ptr
Definition Model.hpp:31

CLI
Definition Benchmark.hpp:21

ppforest2::cli
Command-line interface: argument parsing, subcommands, and benchmark/evaluation orchestration.
Definition Benchmark.hpp:25

ppforest2::cli::setup_train
void setup_train(CLI::App &app, Params &params)
Register train subcommand options on app.

ppforest2::cli::run_train
int run_train(Params &params)
Run the train subcommand.

ppforest2::cli::add_model_options
void add_model_options(CLI::App *sub, ModelParams &model)
Add shared model options (size, lambda, threads, seed, vars) to sub.

ppforest2::cli::train_model
TrainResult train_model(ppforest2::types::FeatureMatrix const &x, ppforest2::types::OutcomeVector const &y, Params const &params)
Train a single model (Forest or Tree) on the given dataset.

ppforest2::types::FeatureMatrix
Eigen::Matrix< Feature, Eigen::Dynamic, Eigen::Dynamic > FeatureMatrix
Dynamic-size matrix of feature values.
Definition Types.hpp:33

ppforest2::types::OutcomeVector
Eigen::Matrix< Outcome, Eigen::Dynamic, 1 > OutcomeVector
Dynamic-size column vector of predictions.
Definition Types.hpp:42

ppforest2::cli::ModelParams
Model training parameters shared by train and evaluate.
Definition ModelParams.hpp:14

ppforest2::cli::Params
All CLI options and runtime parameters.
Definition CLIOptions.hpp:33

ppforest2::cli::TrainResult
Result of a train operation containing the model and training duration.
Definition Train.hpp:27

ppforest2::cli::TrainResult::duration
long long duration
Definition Train.hpp:29

ppforest2::cli::TrainResult::model
ppforest2::Model::Ptr model
Definition Train.hpp:28