diff --git a/tensorflow_ranking/examples/tf_ranking_libsvm.py b/tensorflow_ranking/examples/tf_ranking_libsvm.py index d244e26..3fd44a3 100644 --- a/tensorflow_ranking/examples/tf_ranking_libsvm.py +++ b/tensorflow_ranking/examples/tf_ranking_libsvm.py @@ -118,7 +118,7 @@ def load_libsvm_data(path, list_size): def _parse_line(line): """Parses a single line in LibSVM format.""" - tokens = line.split() + tokens = line.split("#")[0].split() assert len(tokens) >= 2, "Ill-formatted line: {}".format(line) label = float(tokens[0]) qid = tokens[1]