Skip to content

add config examples to use PQ and SQ indexing and search for wiki-1M with Cohere embeddings#1047

Open
harsha-simhadri wants to merge 1 commit intomainfrom
harshasi/add_wiki1M_examples
Open

add config examples to use PQ and SQ indexing and search for wiki-1M with Cohere embeddings#1047
harsha-simhadri wants to merge 1 commit intomainfrom
harshasi/add_wiki1M_examples

Conversation

@harsha-simhadri
Copy link
Copy Markdown
Contributor

add config examples to use PQ and SQ indexing and search for wiki-1M with Cohere embeddings

Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds new benchmark configuration examples intended to demonstrate product quantization (PQ) and spherical quantization workflows for the wikipedia-1M Cohere embedding dataset.

Changes:

  • Added a PQ graph-index build+search example JSON for wikipedia-1M.
  • Added an exhaustive spherical-quantization example JSON (currently named as a wiki1M graph-index example).

Reviewed changes

Copilot reviewed 1 out of 2 changed files in this pull request and generated 1 comment.

File Description
diskann-benchmark/example/graph-index-spherical-quantization-wiki1M.json Adds an exhaustive spherical quantization benchmark config (but currently references siftsmall test data / exhaustive tag).
diskann-benchmark/example/graph-index-product-quantization-wiki1M.json Adds a PQ graph-index benchmark config for wikipedia-1M (currently uses an unregistered job type).

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

"../big-ann-benchmarks/data/wikipedia_cohere" ],
"jobs": [
{
"type": "async-index-build-pq",
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants