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 |
|