Qifan Zhang
commited on
Commit
·
f2c67c4
1
Parent(s):
edd4815
fix p1_flexibility bug
Browse files- utils/pipeline.py +2 -2
utils/pipeline.py
CHANGED
|
@@ -32,10 +32,10 @@ def p1_flexibility(df: pd.DataFrame, model_name: str) -> pd.DataFrame:
|
|
| 32 |
count = 0
|
| 33 |
score = 0
|
| 34 |
for i in range(len(responses_vec)):
|
| 35 |
-
for j in range(
|
| 36 |
if i == j:
|
| 37 |
continue
|
| 38 |
-
score += cos_sim(responses_vec[i], responses_vec[j]).item()
|
| 39 |
count += 1
|
| 40 |
return score / count
|
| 41 |
|
|
|
|
| 32 |
count = 0
|
| 33 |
score = 0
|
| 34 |
for i in range(len(responses_vec)):
|
| 35 |
+
for j in range(len(responses_vec)):
|
| 36 |
if i == j:
|
| 37 |
continue
|
| 38 |
+
score += 1 - cos_sim(responses_vec[i], responses_vec[j]).item()
|
| 39 |
count += 1
|
| 40 |
return score / count
|
| 41 |
|