search

tensorflow / neural-structured-learning

Training neural models with structured signals.
https://www.tensorflow.org/neural_structured_learning
Apache License 2.0
980 stars 189 forks source link

An example of similarity search #89

Closed sayakpaul closed 3 years ago

sayakpaul commented 3 years ago

I was wondering if the team would be interested to work on a tutorial to show how NSL could be used for similarity and retrieval tasks using the graph APIs provided by NSL. Would be even interesting to benchmark the performance of that against other nearest neighbors' approaches.

I personally think it would be helpful for many developers working in this area.

arjung commented 3 years ago

NSL currently does not have APIs to retrieve nearest neighbors. It has APIs to build a graph based on dense embeddings using locality sensitive hashing. You might be interested in https://github.com/google-research/google-research/tree/master/scann.