Skip to content

Commit ec47c61

Browse files
committed
Fix kaggle submission format in format_lines()
1 parent c04fd1f commit ec47c61

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

inference.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def format_lines(video_ids, predictions, top_k):
8383
for class_index in top_indices]
8484
line = sorted(line, key=lambda p: -p[1])
8585
yield video_ids[video_index].decode('utf-8') + "," + " ".join(
86-
"%i:%g" % (label, score) for (label, score) in line) + "\n"
86+
"%i %g" % (label, score) for (label, score) in line) + "\n"
8787

8888

8989
def get_input_data_tensors(reader, data_pattern, batch_size, num_readers=1):

0 commit comments

Comments
 (0)