CLIOptions.hpp

Go to the documentation of this file.

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

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

  struct Params {
    Subcommand subcommand = Subcommand::none;
 
    ModelParams model;
    SimulateParams simulation;
    EvaluateParams evaluate;
    BenchmarkParams benchmark;
    ServeParams serve;
 
    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;
 
    Params() = default;

    explicit Params(nlohmann::json const& config);

    void resolve();

    void resolve_seed();

    stats::DataPacket read_data(stats::RNG& rng);

    void resolve_defaults(unsigned int total_vars);

    nlohmann::json to_json() const;
  };

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

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