Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enable PECOS HNSW with command line interface #79

Merged
merged 1 commit into from
Oct 18, 2021

Conversation

OctoberChang
Copy link
Contributor

Issue #, if available:

Description of changes:
Enable PECOS HNSW with command line interface such that

  • pecos.ann.hnsw.train: train and save the HNSW index
  • pecos.ann.hnsw.predict: load the HNSW index and inference on provided queries

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

```
where `Yt_pred` is a `scipy.sparse.csr_matrix` whose row indices are sorted by its distances ascendingly.
Copy link
Contributor

Choose a reason for hiding this comment

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

I think the nonzero column indices for each row is sorted by its distances in a n ascending order.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

fixed in next revision.

@OctoberChang OctoberChang merged commit 6b85049 into amzn:mainline Oct 18, 2021
@OctoberChang OctoberChang deleted the hnsw-branch branch October 19, 2021 21:04
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