Skip to content

Latest commit

 

History

History
31 lines (20 loc) · 1.27 KB

README.md

File metadata and controls

31 lines (20 loc) · 1.27 KB

Archon Evaluation on GSM8K

Generate answers

First, you will have to generate your answers for the Grade School Math 8K (GSM8K) dataset using gen_answers.py from the archon subdirectory (the parent folder of this folder).

An example of how to do that will be to run this script.

python3 gen_answers.py --config configs/<your-config-file>.json --benchmark gsm8k --parallel 8

If the name of the model in configs/<your-config-file>.json is <model_name>, this will save your output as outputs/gsm8k/model_answer/<model_name>.json folder in the archon subdirectory.

Evaluate answers

After you have your answers, you can run this script to evaluate their correctness by running this script from the archon subdirectory:

python3 gsm8k/gsm8k_evaluation.py --answers-json-path <output_file_name>

This will directly print the accuracy, and the number of problematic queries to the terminal.

Complete Workflow.

Within archon/, you can run

python3 gen_answers.py --config configs/archon-70Bx10_1_samples_then_fuser_with_Qwen2_72B.json --benchmark gsm8k --parallel 8

python3 gsm8k/gsm8k_evaluation.py --answers-json-path outputs/gsm8k/model_answer/archon-70Bx10_1_samples_then_fuser_with_Qwen2_72B.json