pbt_trial.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
#include "pulsatrix/search_space.hpp"
#include "pulsatrix/successive_halving.hpp"
 
namespace pulsatrix {
 
class PBTResumableTrial : public ResumableTrial {
public:
    ~PBTResumableTrial() override = default;
 
    [[nodiscard]] virtual std::vector<double> GetWeights() const = 0;
 
    virtual void SetWeights(const std::vector<double>& weights) = 0;
 
    [[nodiscard]] virtual Configuration GetHyperparameters() const = 0;
 
    virtual void SetHyperparameters(const Configuration& config) = 0;
};
 
}  // namespace pulsatrix