kovacsvi commited on
Commit
3cc9ea6
·
1 Parent(s): 73ee479

debug print for minor probs

Browse files
Files changed (1) hide show
  1. interfaces/cap_minor_media.py +2 -0
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