Method Train

src/algorithm/ivf.cpp:355–375

}

void
IVF::Train(const DatasetPtr& data) {
    if (this->is_trained_) {
        return;
    }

    int64_t total_elements = data->GetNumElements();
    int64_t dim = data->GetDim();
    DatasetPtr train_data =
        vsag::sample_train_data(data, total_elements, dim, train_sample_count_, allocator_);
    int64_t sample_count = train_data->GetNumElements();

    partition_strategy_->Train(train_data);

    const auto* data_ptr = train_data->GetFloat32Vectors();
    this->bucket_->Train(data_ptr, sample_count);
    if (use_reorder_) {
        this->reorder_codes_->Train(data->GetFloat32Vectors(), data->GetNumElements());
    }
    this->is_trained_ = true;
}

std::vector<int64_t>
IVF::Add(const DatasetPtr& base) {