knn retrieval evaluation

[GentleZhu]

Issue #, if available:

Description of changes: This is a draft PR because I found the recall@K is extremly low and it seems the remap get wrong from reading embeddings.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.