load_kmknn_prebuilt.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_KMKNN_LOAD_KMKNN_PREBUILT_HPP
#define KNNCOLLE_KMKNN_LOAD_KMKNN_PREBUILT_HPP
 
#include "knncolle/knncolle.hpp"
 
#include <string>
 
namespace knncolle_kmknn {
 
struct KmknnPrebuiltTypes {
    knncolle::NumericType kmeansfloat;
};
 
inline KmknnPrebuiltTypes load_kmknn_prebuilt_types(const std::filesystem::path& dir) {
    knncolle::NumericType type;
    knncolle::quick_load(dir / "FLOAT_TYPE", &type, 1);
 
    KmknnPrebuiltTypes config;
    config.kmeansfloat = type;
 
    return config;
}
 
template<
    typename Index_,
    typename Data_,
    typename Distance_,
    class DistanceMetricData_ = knncolle::DistanceMetric<Data_, Distance_>,
    typename KmeansFloat_ = Distance_,
    class DistanceMetricCenter_ = knncolle::DistanceMetric<KmeansFloat_, Distance_>
>
auto load_kmknn_prebuilt(const std::filesystem::path& dir) {
    return new KmknnPrebuilt<
        Index_,
        Data_,
        Distance_,
        DistanceMetricData_,
        KmeansFloat_,
        DistanceMetricCenter_
    >(dir);
}
 
}
 
#endif