This commit is contained in:
ysymyth 2023-07-03 22:39:33 -04:00
parent 23629f6da3
commit 066015df56

View File

@ -39,10 +39,17 @@ from tot.tasks.game24 import Game24Task
args = argparse.Namespace(backend='gpt-4', temperature=0.7, task='game24', naive_run=False, prompt_sample=None, method_generate='propose', method_evaluate='value', method_select='greedy', n_generate_sample=1, n_evaluate_sample=3, n_select_sample=5)
task = Game24Task()
solve(args, task, 900)
ys, infos = solve(args, task, 900)
print(ys[0])
```
And the output would be something like (note it's not deterministic, and sometimes the output can be wrong):
```
10 - 4 = 6 (left: 5 6 6)
5 * 6 = 30 (left: 6 30)
30 - 6 = 24 (left: 24)
Answer: (5 * (10 - 4)) - 6 = 24
```
## Paper Experiments