poltextlab commited on
Commit
829404f
·
verified ·
1 Parent(s): 2490733

update ordering

Browse files
Files changed (1) hide show
  1. interfaces/cap_minor.py +1 -1
interfaces/cap_minor.py CHANGED
@@ -79,7 +79,7 @@ def predict(text, model_id, tokenizer_id):
79
  output_pred = {f"[{'999' if str(CAP_MIN_NUM_DICT[i]) == '999' else str(CAP_MIN_NUM_DICT[i])[:-2]}]{convert_minor_to_major(CAP_MIN_NUM_DICT[i])} [{CAP_MIN_NUM_DICT[i]}]{CAP_MIN_LABEL_NAMES[CAP_MIN_NUM_DICT[i]]}": probs[i] for i in np.argsort(probs)[::-1]}
80
 
81
 
82
- output_pred = dict(sorted(output_pred.items(), key=lambda item: item[1]))
83
  first_n_items = take(5, output_pred.items())
84
 
85
  html = ""
 
79
  output_pred = {f"[{'999' if str(CAP_MIN_NUM_DICT[i]) == '999' else str(CAP_MIN_NUM_DICT[i])[:-2]}]{convert_minor_to_major(CAP_MIN_NUM_DICT[i])} [{CAP_MIN_NUM_DICT[i]}]{CAP_MIN_LABEL_NAMES[CAP_MIN_NUM_DICT[i]]}": probs[i] for i in np.argsort(probs)[::-1]}
80
 
81
 
82
+ output_pred = dict(sorted(output_pred.items(), key=lambda item: item[1], reverse=True))
83
  first_n_items = take(5, output_pred.items())
84
 
85
  html = ""