import json from reranker import rankingfunc from sample_data import BASIC_EXAMPLE def main(): print("Initial") print(json.dumps(BASIC_EXAMPLE, indent=4)) k=10 print("Re-ranking top {}".format(k)) result = rankingfunc(BASIC_EXAMPLE,debug=True,k=k) print(json.dumps(result, indent=4)) if __name__ == "__main__": main()