add an interface for predicting top-k results