Merge pull request #5 from jplu:master

PiperOrigin-RevId: 224409654
tensorflow · Dec 6, 2018 · 931e4e1 · 931e4e1
2 parents ad34142 + f539777
commit 931e4e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]