fixed sorting predictions by probabilities in wrong order

This commit is contained in:
Matthias Bisping 2022-03-30 01:14:03 +02:00
parent 95209a5c9d
commit ad6bb80900

View File

@ -36,7 +36,7 @@ class Classifier:
)
def __format_array_prediction_format(self, prediction):
cls2prob = dict(sorted(zip(self._classes, prediction), key=itemgetter(1)))
cls2prob = dict(sorted(zip(self._classes, prediction), key=itemgetter(1), reverse=True))
most_likely = [*cls2prob][0]
return {"label": most_likely, "probabilities": cls2prob}