Spaces:
Sleeping
Sleeping
kovacsvi
commited on
Commit
·
18797a3
1
Parent(s):
ba0fb9c
removed filtering logic
Browse files
app.py
CHANGED
|
@@ -176,17 +176,9 @@ def plot_average_emotion_barplot(heatmap_data):
|
|
| 176 |
labels = [id2label[i] for i in range(len(mean_scores))]
|
| 177 |
scores = mean_scores
|
| 178 |
|
| 179 |
-
# Optional: filter out near-zero average emotions
|
| 180 |
-
labels_filtered = []
|
| 181 |
-
scores_filtered = []
|
| 182 |
-
for l, s in zip(labels, scores):
|
| 183 |
-
if s > 0.01:
|
| 184 |
-
labels_filtered.append(l)
|
| 185 |
-
scores_filtered.append(s)
|
| 186 |
-
|
| 187 |
# Sort for better visualization
|
| 188 |
sorted_data = sorted(
|
| 189 |
-
zip(labels_filtered,
|
| 190 |
)
|
| 191 |
sorted_labels, sorted_scores = zip(*sorted_data)
|
| 192 |
|
|
|
|
| 176 |
labels = [id2label[i] for i in range(len(mean_scores))]
|
| 177 |
scores = mean_scores
|
| 178 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 179 |
# Sort for better visualization
|
| 180 |
sorted_data = sorted(
|
| 181 |
+
zip(labels_filtered, scores), key=lambda x: x[1], reverse=True
|
| 182 |
)
|
| 183 |
sorted_labels, sorted_scores = zip(*sorted_data)
|
| 184 |
|