Milad Alshomary
commited on
Commit
·
dcbbcbd
1
Parent(s):
51ad242
updates
Browse files
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 |
-
|
|
|
|
| 615 |
else:
|
| 616 |
for feature, spans in feature_map.items():
|
| 617 |
-
|
|
|
|
| 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 |
|