UniqAI/tree-of-thought-llm
1
0
Fork 0
mirror of https://github.com/princeton-nlp/tree-of-thought-llm synced 2024-11-24 21:14:22 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

modify value_outputs_unwrap method

Browse Source
This commit is contained in:
whatchang 2024-02-12 13:42:04 +09:00
parent ab400345c5
commit 62d05429c7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tot/tasks/game24.py
View File

@ -88,5 +88,5 @@ class Game24Task(Task):
return 0 return 0
value_names = [_.split('\n')[-1] for _ in value_outputs] value_names = [_.split('\n')[-1] for _ in value_outputs]
value_map = {'impossible': 0.001, 'likely': 1, 'sure': 20} # TODO: ad hoc value_map = {'impossible': 0.001, 'likely': 1, 'sure': 20} # TODO: ad hoc
value = sum(value * value_names.count(name) for name, value in value_map.items()) value = sum(value * target_value.count(name) for name, value in value_map.items() for target_value in value_names)
return value return value