Exact Vector Reranking

Medium · Solved in PyTorch · Machine learning coding practice

Problem

Rerank the one-dimensional candidate_ids using vectors of shape (N,D) and query of shape (D,):

Remove duplicate candidate IDs.

Topics: vector-search, reranking, indexing

The full statement, worked examples, hints and the test suite are available once you sign in. You can then solve Exact Vector Reranking in the browser and run it against the tests.

Related problems