lhzstar commited on
Commit
fe183af
1 Parent(s): 93c37ff

new commits

Browse files
Files changed (1) hide show
  1. run_eval.py +1 -1
run_eval.py CHANGED
@@ -15,7 +15,7 @@ SENTTR_MODEL_ID = "sentence-transformers/all-mpnet-base-v2"
15
  celeb_names = ["Cate Blanchett", "David Beckham", "Emma Watson", "Lady Gaga", "Madonna", "Mark Zuckerberg"]
16
 
17
  USE_UNLIMIFORMER = True
18
- TOP_K = 8
19
 
20
  celeb_data = get_celeb_data("data.json")
21
  references = [val['answers'] for key, val in list(celeb_data.items()) if key in celeb_names]
 
15
  celeb_names = ["Cate Blanchett", "David Beckham", "Emma Watson", "Lady Gaga", "Madonna", "Mark Zuckerberg"]
16
 
17
  USE_UNLIMIFORMER = True
18
+ TOP_K = 16
19
 
20
  celeb_data = get_celeb_data("data.json")
21
  references = [val['answers'] for key, val in list(celeb_data.items()) if key in celeb_names]