TrainingSpec.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/PPPDAStrategy.hpp"
#include "models/DRUniformStrategy.hpp"
#include "models/DRNoopStrategy.hpp"
#include "models/SRMeanOfMeansStrategy.hpp"
 
#include <memory>
#include <thread>
#include <nlohmann/json.hpp>
 
namespace ppforest2 {
  struct TrainingSpec {
    using Ptr = std::shared_ptr<TrainingSpec>;

    pp::PPStrategy::Ptr const pp_strategy;
    dr::DRStrategy::Ptr const dr_strategy;
    sr::SRStrategy::Ptr const sr_strategy;

    int const size;
    int const seed;
    int const threads;
    int const max_retries;

    TrainingSpec(pp::PPStrategy::Ptr pp,
                 dr::DRStrategy::Ptr dr,
                 sr::SRStrategy::Ptr sr,
                 int size        = 0,
                 int seed        = 0,
                 int threads     = 0,
                 int max_retries = 3);

    bool is_forest() const { return size > 0; }

    void to_json(nlohmann::json& j) const;

    static Ptr from_json(nlohmann::json const& j);

    template<typename... Args> static Ptr make(Args&&... args) {
      return std::make_shared<TrainingSpec>(std::forward<Args>(args)...);
    }

    int resolve_threads() const { return threads > 0 ? threads : std::thread::hardware_concurrency(); }
  };
}