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.5034 | ± | 0.0146 |
| none | 0 | acc_norm | ↑ | 0.5068 | ± | 0.0146 | ||
| gsm8k_platinum_cot | 3 | flexible-extract | 8 | exact_match | ↑ | 0.7601 | ± | 0.0123 |
| strict-match | 8 | exact_match | ↑ | 0.5211 | ± | 0.0144 |