Views
No views yet
1# eval
2from lm_eval.tasks import TaskManager
3from lm_eval.utils import make_table
4
5with tempfile.TemporaryDirectory() as tmp_dir:
6 results = GPTQModel.eval(
7 QUANT_SAVE_PATH,
8 tasks=[EVAL.LM_EVAL.ARC_CHALLENGE, EVAL.LM_EVAL.GSM8K_PLATINUM_COT],
9 apply_chat_template=True,
10 random_seed=898,
11 output_path= tmp_dir,
12 )
13
14 print(make_table(results))
15 if "groups" in results:
16 print(make_table(results, "groups"))| Tasks | Version | Filter | n-shot | Metric | Value | Stderr | ||
|---|---|---|---|---|---|---|---|---|
| arc_challenge | 1 | none | 0 | acc | ↑ | 0.5000 | ± | 0.0146 |
| none | 0 | acc_norm | ↑ | 0.5128 | ± | 0.0146 | ||
| gsm8k_platinum_cot | 3 | flexible-extract | 8 | exact_match | ↑ | 0.3995 | ± | 0.0141 |
| strict-match | 8 | exact_match | ↑ | 0.2548 | ± | 0.0125 |