Qifan Zhang commited on
Commit
f2c67c4
1 Parent(s): edd4815

fix p1_flexibility bug

Browse files
Files changed (1) hide show
  1. 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(1, len(responses_vec)):
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