poltextlab commited on
Commit
5c48f48
·
verified ·
1 Parent(s): a552e92

handle 999 case

Browse files
Files changed (1) hide show
  1. interfaces/cap_minor.py +4 -0
interfaces/cap_minor.py CHANGED
@@ -37,6 +37,10 @@ def convert_minor_to_major(results, probs):
37
  for i in results:
38
  prob = probs[i]
39
  major_code = str(CAP_MIN_NUM_DICT[i])[:-2]
 
 
 
 
40
  label = CAP_LABEL_NAMES[int(major_code)]
41
 
42
  key = f"[{major_code}] {label}"
 
37
  for i in results:
38
  prob = probs[i]
39
  major_code = str(CAP_MIN_NUM_DICT[i])[:-2]
40
+
41
+ if major_code == "99":
42
+ major_code = "999"
43
+
44
  label = CAP_LABEL_NAMES[int(major_code)]
45
 
46
  key = f"[{major_code}] {label}"