Use more efficient KNN methods for link prediction

[zheng-da]

We should implement a proper link prediction inference script. Users can provide a list of nodes and the link prediction inference script should return top K nodes that are likely to connect with the input nodes. We should use FAISS for link prediction.