joshnguyen commited on
Commit
1d1d7a2
·
1 Parent(s): 508ada6

Update sorting mechanism

Browse files
Files changed (1) hide show
  1. custom_label.py +2 -2
custom_label.py CHANGED
@@ -115,11 +115,11 @@ class CustomLabel(Changeable, Selectable, IOComponent, JSONSerializable):
115
  y = y["confidences"]
116
  y = {c["label"]: c["confidence"] for c in y}
117
  sorted_pred = sorted(
118
- y.items(), key=operator.itemgetter(0), reverse=True)
119
  if self.num_top_classes is not None:
120
  sorted_pred = sorted_pred[: self.num_top_classes]
121
  return {
122
- "label": sorted_pred[0][0],
123
  "confidences": [
124
  {"label": pred[0], "confidence": pred[1]} for pred in sorted_pred
125
  ],
 
115
  y = y["confidences"]
116
  y = {c["label"]: c["confidence"] for c in y}
117
  sorted_pred = sorted(
118
+ y.items(), key=operator.itemgetter(0), reverse=False)
119
  if self.num_top_classes is not None:
120
  sorted_pred = sorted_pred[: self.num_top_classes]
121
  return {
122
+ "label": None,
123
  "confidences": [
124
  {"label": pred[0], "confidence": pred[1]} for pred in sorted_pred
125
  ],