EvaluateParams.hpp

Go to the documentation of this file.

#pragma once
 
#include "stats/Simulation.hpp"
 
#include <nlohmann/json.hpp>
#include <optional>
#include <string>
#include <vector>
 
namespace ppforest2::cli {
  struct SimulateParams {
    SimulateParams() = default;
 
    std::string format;
    int rows     = 1000;
    int cols     = 10;
    int n_groups = 2;
 
    stats::simulation::params::Classification classification;
    stats::simulation::params::Regression regression;

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

    void resolve_format();

    stats::DataPacket simulate(types::Mode mode, stats::RNG& rng) const;
  };

  struct EvaluateParams {
    EvaluateParams() = default;
 
    std::optional<float> train_ratio;
    std::optional<int> iterations; 
    int warmup = 0;                
    bool fixed_seed = false;
    std::string export_path;

    bool convergence_enabled() const { return !iterations || *iterations <= 0; }

    struct Convergence {
      std::optional<float> cv;   
      std::optional<int> min;    
      std::optional<int> window; 
      std::optional<int> max;    
    } convergence;

    static void validate(nlohmann::json const& config, std::vector<std::string>& errors);

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

    void resolve_defaults();

    nlohmann::json to_json() const;

    nlohmann::json overrides() const;
  };
}