Spaces:
Running
Running
kovacsvi
commited on
Commit
·
6577743
1
Parent(s):
658551c
sort is in-place
Browse files
interfaces/cap_minor_media.py
CHANGED
@@ -42,7 +42,8 @@ major_to_minor_map = dict(major_to_minor_map)
|
|
42 |
def normalize_probs(probs: dict, n: int):
|
43 |
probs = list(probs.values())
|
44 |
if len(probs) > n:
|
45 |
-
probs
|
|
|
46 |
values = np.array(probs)
|
47 |
exp_values = np.exp(values)
|
48 |
sum_exp = np.sum(exp_values)
|
|
|
42 |
def normalize_probs(probs: dict, n: int):
|
43 |
probs = list(probs.values())
|
44 |
if len(probs) > n:
|
45 |
+
probs.sort(reverse=True)
|
46 |
+
probs = probs[:5]
|
47 |
values = np.array(probs)
|
48 |
exp_values = np.exp(values)
|
49 |
sum_exp = np.sum(exp_values)
|