Spaces:
Running
Running
kovacsvi
commited on
Commit
·
3cc9ea6
1
Parent(s):
73ee479
debug print for minor probs
Browse files
interfaces/cap_minor_media.py
CHANGED
@@ -124,7 +124,9 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
|
|
124 |
# Restrict to valid minor codes
|
125 |
valid_indices = [minor_id_to_index[mid] for mid in valid_minor_ids if mid in minor_id_to_index]
|
126 |
filtered_probs = {minor_index_to_id[i]: float(minor_probs[0][i]) for i in valid_indices}
|
|
|
127 |
filtered_probs = normalize_probs(filtered_probs)
|
|
|
128 |
|
129 |
output_pred = {
|
130 |
f"[{top_major_id}] {CAP_MEDIA_LABEL_NAMES[top_major_id]} [{k}] {CAP_MIN_LABEL_NAMES[k]}": v
|
|
|
124 |
# Restrict to valid minor codes
|
125 |
valid_indices = [minor_id_to_index[mid] for mid in valid_minor_ids if mid in minor_id_to_index]
|
126 |
filtered_probs = {minor_index_to_id[i]: float(minor_probs[0][i]) for i in valid_indices}
|
127 |
+
print(filtered_probs) # debug
|
128 |
filtered_probs = normalize_probs(filtered_probs)
|
129 |
+
print(filtered_probs) # debug
|
130 |
|
131 |
output_pred = {
|
132 |
f"[{top_major_id}] {CAP_MEDIA_LABEL_NAMES[top_major_id]} [{k}] {CAP_MIN_LABEL_NAMES[k]}": v
|