From c757b838dd3cca874437cb4b229add91fe7afbf7 Mon Sep 17 00:00:00 2001 From: marcsello Date: Mon, 14 Jun 2021 03:28:32 +0200 Subject: [PATCH] Fixed float conversion --- cnn_classification_service/cnn_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cnn_classification_service/cnn_classifier.py b/cnn_classification_service/cnn_classifier.py index 47e9201..083ed02 100644 --- a/cnn_classification_service/cnn_classifier.py +++ b/cnn_classification_service/cnn_classifier.py @@ -77,7 +77,7 @@ class Classifier(object): } labels = dict((v, k) for k, v in labels.items()) - labeled_predictions = {labels[i]: p for i, p in enumerate(prediction[0])} + labeled_predictions = {labels[i]: float(p) for i, p in enumerate(prediction[0])} predicted_class_name = [labels[k] for k in predicted_class_indices][0] # eh?