Milad Alshomary commited on
Commit
dcbbcbd
·
1 Parent(s): 51ad242
Files changed (1) hide show
  1. utils/interp_space_utils.py +4 -2
utils/interp_space_utils.py CHANGED
@@ -611,10 +611,12 @@ def compute_clusters_style_representation_3(
611
  for author, feature_map in spans_by_author.items():
612
  if author in task_author_names.intersection(set(cluster_ids)):
613
  for feature, spans in feature_map.items():
614
- feature_importance[feature] += len(spans)
 
615
  else:
616
  for feature, spans in feature_map.items():
617
- feature_importance[feature] -= len(spans)
 
618
  print(feature_importance)
619
  selected_features_ranked = sorted(feature_importance, key=lambda f: -feature_importance[f])[:int(top_k)]
620
 
 
611
  for author, feature_map in spans_by_author.items():
612
  if author in task_author_names.intersection(set(cluster_ids)):
613
  for feature, spans in feature_map.items():
614
+ if spans:
615
+ feature_importance[feature] += len(spans)
616
  else:
617
  for feature, spans in feature_map.items():
618
+ if spans:
619
+ feature_importance[feature] -= len(spans)
620
  print(feature_importance)
621
  selected_features_ranked = sorted(feature_importance, key=lambda f: -feature_importance[f])[:int(top_k)]
622