Function register_ivf

python_bindings/ivf_bindings.cpp:151–181

Register IVF bindings into combined module

        if (!built_) {
            throw std::runtime_error("IvfIndex must be built or loaded before save");
        }
        index_->save(path.c_str());
    }

    static IvfIndex load(const std::string& path) {
        IvfIndex wrapper;
        wrapper.index_ = std::make_unique<rabitqlib::ivf::IVF>();
        wrapper.index_->load(path.c_str());
        wrapper.dim_ = wrapper.index_->dimension();
        wrapper.max_elements_ = wrapper.index_->max_elements();
        wrapper.num_clusters_ = wrapper.index_->num_clusters();
        wrapper.nbits_ = wrapper.index_->nbits();
        wrapper.metric_ = wrapper.index_->metric_type();
        wrapper.built_ = true;
        return wrapper;
    }

    [[nodiscard]] size_t dim() const { return dim_; }
    [[nodiscard]] size_t max_elements() const { return max_elements_; }
    [[nodiscard]] size_t num_clusters() const { return num_clusters_; }
    [[nodiscard]] size_t nbits() const { return nbits_; }
    [[nodiscard]] bool is_built() const { return built_; }
    [[nodiscard]] std::string metric() const { return metric_to_string(metric_); }

   private:
    IvfIndex() = default;

    size_t dim_ = 0;
    size_t max_elements_ = 0;
    size_t num_clusters_ = 0;
    size_t nbits_ = 0;
    rabitqlib::MetricType metric_ = rabitqlib::METRIC_L2;
    bool built_ = false;
    std::unique_ptr<rabitqlib::ivf::IVF> index_;