ModelParams.hpp

Go to the documentation of this file.

#pragma once
 
#include <nlohmann/json.hpp>
#include <optional>
#include <string>
#include <vector>
 
namespace ppforest2::cli {
  struct ModelParams {
    ModelParams() = default;
 
    int size     = 100;
    float lambda = 0.0;
    std::optional<int> threads;
    std::optional<int> seed;
    std::optional<float> p_vars;
    std::optional<int> n_vars;
    int max_retries = 3;
    std::string p_vars_input;

    std::string mode_input;

    std::string pp_input;
    std::string vars_input;
    std::string cutpoint_input;
    std::vector<std::string> stop_inputs;
    std::string binarize_input;
    std::string grouping_input;
    std::string leaf_input;

    nlohmann::json pp_config;
    nlohmann::json vars_config;
    nlohmann::json cutpoint_config;
    nlohmann::json stop_config;
    nlohmann::json binarize_config;
    nlohmann::json grouping_config;
    nlohmann::json leaf_config;

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

    void resolve();

    void resolve_defaults(unsigned int total_vars);

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

    nlohmann::json to_json() const;
  };

  float parse_proportion(std::string const& input);
  float parse_proportion(nlohmann::json const& j);

  nlohmann::json strategy_string_to_json(std::string const& input);
}