Commit 77b652ca authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[utils] revert map_labels to original

parent 0980593a
Pipeline #26376 failed with stage
in 5 minutes and 52 seconds
......@@ -90,7 +90,8 @@ def map_labels(raw_labels, start_index=0):
for i in range(len(possible_labels)):
l = possible_labels[i]
labels[numpy.where(labels==l)[0][0]] = i + start_index
#labels[numpy.where(labels==l)[0][0]] = i + start_index
labels[numpy.where(labels==l)[0]] = i + start_index
# -----
# map back to native int, resolve the problem with dataset concatenation
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment