pminervini commited on
Commit
5cdba42
1 Parent(s): f69201c
Files changed (1) hide show
  1. src/display/utils.py +5 -3
src/display/utils.py CHANGED
@@ -31,13 +31,15 @@ class Tasks(Enum):
31
  truthfulqa_mc2 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2")
32
 
33
  halueval_qa = Task("halueval_qa", "acc", "HaluEval QA")
 
 
34
 
35
- # xsum = Task("xsum", "factKB", "XSum")
36
- # cnndm = Task("cnndm", "factKB", "CNN/DM")
37
 
38
  memotrap = Task("memo-trap", "acc", "MemoTrap")
39
 
40
- # ifeval = Task("ifeval", "prompt_level_strict_acc", "IFEval")
41
 
42
  #truthfulqa_mc1 = Task("truthfulqa_mc1", "acc", "TruthfulQA MC1")
43
  #truthfulqa_mc2 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2")
 
31
  truthfulqa_mc2 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2")
32
 
33
  halueval_qa = Task("halueval_qa", "acc", "HaluEval QA")
34
+ halueval_qa = Task("halueval_summarization", "acc", "HaluEval Summ")
35
+ halueval_qa = Task("halueval_dialogue", "acc", "HaluEval Dial")
36
 
37
+ xsum = Task("xsum", "factKB", "XSum")
38
+ cnndm = Task("cnndm", "factKB", "CNN/DM")
39
 
40
  memotrap = Task("memo-trap", "acc", "MemoTrap")
41
 
42
+ ifeval = Task("ifeval", "prompt_level_strict_acc", "IFEval")
43
 
44
  #truthfulqa_mc1 = Task("truthfulqa_mc1", "acc", "TruthfulQA MC1")
45
  #truthfulqa_mc2 = Task("truthfulqa_mc2", "acc", "TruthfulQA MC2")