CLIOptions.hpp

Go to the documentation of this file.

#pragma once
 
#include "cli/ModelParams.hpp"
#include "cli/EvaluateParams.hpp"
#include "cli/BenchmarkParams.hpp"
#include "io/Output.hpp"
 
#include <nlohmann/json.hpp>
#include <string>

namespace ppforest2::cli {
  enum class Subcommand { none, train, predict, evaluate, benchmark, summarize };

  struct CLIOptions {
    Subcommand subcommand = Subcommand::none;
 
    ModelParams model;
    SimulateParams simulation;
    ConvergenceParams convergence;
    EvaluateParams evaluate;
    BenchmarkParams benchmark;
 
    std::string data_path;
    std::string save_path = "model.json";
    std::string model_path;
    std::string output_path;
 
    bool quiet          = false;
    bool no_save        = false;
    bool no_metrics     = false;
    bool no_color       = false;
    bool no_proportions = false;

    std::string config_path;
  };

  void warn_unused_params(io::Output& out, CLIOptions const& params);

  void init_params(CLIOptions& params, int total_vars = 0);

  CLIOptions parse_args(int argc, char* argv[]);
}