Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

fixed sorting predictions by probabilities in wrong order

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-30 01:14:03 +02:00
parent 95209a5c9d
commit ad6bb80900
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
image_prediction/classifier/classifier.py
View File

@ -36,7 +36,7 @@ class Classifier:
) )
def __format_array_prediction_format(self, prediction): 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] most_likely = [*cls2prob][0]
return {"label": most_likely, "probabilities": cls2prob} return {"label": most_likely, "probabilities": cls2prob}