mirror of
https://github.com/princeton-nlp/tree-of-thought-llm
synced 2025-04-28 09:41:09 +00:00
19 lines
692 B
Python
19 lines
692 B
Python
import argparse
|
|
from tot.methods.bfs import solve
|
|
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()
|
|
ys, infos = solve(args=args, task=task, idx=900)
|
|
print(ys[0]) |