selecting top k datapoints with high probablity