From 2c92c94296e66a9180ef288d38e7aff8b31683fd Mon Sep 17 00:00:00 2001
From: Nabeel <khnabeelasif@qlik.com>
Date: Tue, 2 Jun 2020 16:35:18 +1000
Subject: [PATCH] Fix for mapping labels to pretrained models

---
 core/_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/_utils.py b/core/_utils.py
index 448c54c..0e83dd6 100644
--- a/core/_utils.py
+++ b/core/_utils.py
@@ -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)