Fixed float conversion
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/push Build is passing
Details
This commit is contained in:
parent
f752668cb5
commit
c757b838dd
|
@ -77,7 +77,7 @@ class Classifier(object):
|
||||||
}
|
}
|
||||||
labels = dict((v, k) for k, v in labels.items())
|
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?
|
predicted_class_name = [labels[k] for k in predicted_class_indices][0] # eh?
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue