Skip to content

Commit

Permalink
Fix for mapping labels to pretrained models
Browse files Browse the repository at this point in the history
  • Loading branch information
Nabeel committed Jun 2, 2020
1 parent 6afd139 commit 2c92c94
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion core/_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -561,7 +561,7 @@ def decode(y, labels={}):
# Store the shape of the array
shape = y.shape
# Flatten the array and perform the decoding
y = np.array([decode(x, labels) for x in y.ravel()])
y = np.array([decode(atof(x), labels) for x in y.ravel()])
# Reshape the array back to it's original form
y = np.reshape(y, shape)

Expand Down

0 comments on commit 2c92c94

Please sign in to comment.