hpo_genotype.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <algorithm>
#include <cmath>
#include <stdexcept>
#include <vector>
 
#include "pulsatrix/search_space.hpp"
 
namespace pulsatrix {
 
inline Configuration DecodeGenotype(const SearchSpace& space, const std::vector<double>& genotype) {
    if (genotype.size() != space.size()) {
        throw std::invalid_argument("DecodeGenotype: genotype.size() must equal space.size()");
    }
    Configuration config;
    for (size_t i = 0; i < space.parameters().size(); ++i) {
        const auto& spec = space.parameters()[i];
        double gene = genotype[i];
        if (gene < 0.0 || gene > 1.0) {
            throw std::invalid_argument("DecodeGenotype: every gene must be in [0, 1]");
        }
        switch (spec.kind) {
            case ParameterKind::Continuous:
                config[spec.name] = spec.lower + gene * (spec.upper - spec.lower);
                break;
            case ParameterKind::LogUniform: {
                double log_lower = std::log(spec.lower);
                double log_upper = std::log(spec.upper);
                config[spec.name] = std::exp(log_lower + gene * (log_upper - log_lower));
                break;
            }
            case ParameterKind::Integer: {
                double interpolated = spec.lower + gene * (spec.upper - spec.lower);
                int64_t rounded = static_cast<int64_t>(std::round(interpolated));
                rounded = std::min(rounded, static_cast<int64_t>(spec.upper));
                rounded = std::max(rounded, static_cast<int64_t>(spec.lower));
                config[spec.name] = rounded;
                break;
            }
            case ParameterKind::Categorical: {
                size_t num_categories = spec.categories.size();
                size_t index = static_cast<size_t>(gene * static_cast<double>(num_categories));
                index = std::min(index, num_categories - 1);
                config[spec.name] = spec.categories[index];
                break;
            }
        }
    }
    return config;
}
 
}  // namespace pulsatrix