trial.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <optional>
#include <string>
#include <vector>
 
#include "pulsatrix/search_space.hpp"
 
namespace pulsatrix {
 
struct MetricRecord {
    std::string tag;
    double value;
    int step;
};
 
class Trial {
public:
    explicit Trial(Configuration configuration) : configuration_(std::move(configuration)) {}
 
    [[nodiscard]] const Configuration& configuration() const { return configuration_; }
 
    void RecordMetric(std::string tag, double value, int step) {
        metrics_.push_back(MetricRecord{std::move(tag), value, step});
    }
 
    [[nodiscard]] const std::vector<MetricRecord>& metrics() const { return metrics_; }
 
    [[nodiscard]] std::optional<double> LatestMetric(const std::string& tag) const {
        for (auto it = metrics_.rbegin(); it != metrics_.rend(); ++it) {
            if (it->tag == tag) {
                return it->value;
            }
        }
        return std::nullopt;
    }
 
    [[nodiscard]] std::optional<double> BestMetric(const std::string& tag, bool maximize) const {
        std::optional<double> best;
        for (const auto& record : metrics_) {
            if (record.tag != tag) {
                continue;
            }
            if (!best.has_value() || (maximize ? record.value > *best : record.value < *best)) {
                best = record.value;
            }
        }
        return best;
    }
 
private:
    Configuration configuration_;
    std::vector<MetricRecord> metrics_;
};
 
}  // namespace pulsatrix