search

QingyaoAi / Deep-Listwise-Context-Model-for-Ranking-Refinement

A Tensorflow implementation of the Deep Listwise Context Model (DLCM) for ranking refinement.
Apache License 2.0
134 stars 56 forks source link

bug in Prepare_yahoo_letor_data_set1.py #3

Closed wintersurvival closed 5 years ago

wintersurvival commented 5 years ago

Hi QingyaoAi, Thanks for releasing code. In main function of Prepare_yahoo_letor_data_set1.py, the statement: RANK_CUT = sys.argv[4] seems should be modified as: RANK_CUT = int(sys.argv[4]) Otherwise, RANK_CUT is a str, for example '10', its ASCII code is much larger than its int type number.

QingyaoAi commented 5 years ago

Thank you for pointing it out! I have fixed the problem.