pminervini commited on
Commit
8bb4b16
1 Parent(s): 652d88f
Files changed (1) hide show
  1. src/backend/envs.py +2 -1
src/backend/envs.py CHANGED
@@ -22,7 +22,8 @@ class Tasks(Enum):
22
  # task1 = Task("logiqa", "acc_norm", "LogiQA")
23
  task0 = Task("nq_open", "em", "NQ Open", 64) # 64, as in the ATLAS paper
24
  task1 = Task("triviaqa", "em", "TriviaQA", 64) # 64, as in the ATLAS paper
25
- task2 = Task("truthfulqa_mc", "mc2", "TruthfulQA", 0) # TruthfulQA is intended as a zero-shot benchmark [5, 47]. https://owainevans.github.io/pdfs/truthfulQA_lin_evans.pdf
 
26
 
27
 
28
  # NUM_FEWSHOT = 64 # Change with your few shot
 
22
  # task1 = Task("logiqa", "acc_norm", "LogiQA")
23
  task0 = Task("nq_open", "em", "NQ Open", 64) # 64, as in the ATLAS paper
24
  task1 = Task("triviaqa", "em", "TriviaQA", 64) # 64, as in the ATLAS paper
25
+ task2 = Task("truthfulqa_mc1", "mc1", "TruthfulQA MC1", 0)
26
+ task3 = Task("truthfulqa_mc2", "mc2", "TruthfulQA MC2", 0) # TruthfulQA is intended as a zero-shot benchmark [5, 47]. https://owainevans.github.io/pdfs/truthfulQA_lin_evans.pdf
27
 
28
 
29
  # NUM_FEWSHOT = 64 # Change with your few shot