pruner.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <optional>
#include <vector>
 
namespace pineforge::hpo {
 
enum class PrunerKind { None, Median, Halving };
 
class Pruner final {
public:
    Pruner(PrunerKind kind, std::vector<double> rungs, unsigned eta, bool minimize);
    std::vector<std::size_t> bar_counts(std::size_t full_size) const;
    std::vector<std::optional<double>> cuts() const;
    void observe(const std::vector<std::optional<double>>& scores);
    bool prune(double score, std::optional<double> cut) const noexcept;
 
private:
    PrunerKind kind_;
    std::vector<double> rungs_;
    unsigned eta_;
    bool minimize_;
    std::vector<std::vector<double>> history_;
    std::vector<std::size_t> cursors_;
};
 
}  // namespace pineforge::hpo