softmax style normalizaton

interfaces/cap_minor_media.py

@@ -40,14 +40,10 @@ for code in CAP_MIN_CODES:
 major_to_minor_map = dict(major_to_minor_map)
 
 def normalize_probs(probs: dict):
-    min_val = min(probs.values())
-    max_val = max(probs.values())
-    range_val = max_val - min_val
-
-    if range_val == 0:
-        return {k: 1.0 for k in probs}
-
-    return {k: (v - min_val) / range_val for k, v in probs.items()}
+    values = np.array(list(probs.values()))
+    exp_values = np.exp(values)
+    sum_exp = np.sum(exp_values)
+    return {k: float(v) for k, v in zip(probs.keys(), exp_values / sum_exp)}
 
 def check_huggingface_path(checkpoint_path: str):
     try: