smk/graph-of-thoughts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixes #9; Rerun affected document merging results and updated logs (minor improvements for got method)

Browse Source
This commit is contained in:
Nils Blach 2023-10-16 10:42:04 +09:00
parent f6be6c06f2
commit b41d653f21
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/doc_merge/doc_merge.py
View File

@ -219,6 +219,7 @@ NDA <S{num}>:
prompt += self.improve_summary_prompt_end.format(summary=current) prompt += self.improve_summary_prompt_end.format(summary=current)
return prompt return prompt
elif method.startswith("got"): elif method.startswith("got"):
parts = sorted(list(parts)) if len(parts) > 0 else list(range(len(documents)))
if current is None or current == "": if current is None or current == "":
prompt += self.merge_doc_prompt_start.format(num=len(parts)) prompt += self.merge_doc_prompt_start.format(num=len(parts))
for i, part in enumerate(sorted(list(parts))): for i, part in enumerate(sorted(list(parts))):
@ -546,9 +547,7 @@ def got() -> operations.GraphOfOperations:
""" """
operations_graph = operations.GraphOfOperations() operations_graph = operations.GraphOfOperations()
branch_factor = 10 operations_graph.append_operation(operations.Generate(1, 5))
operations_graph.append_operation(operations.Generate(1, branch_factor))
operations_graph.append_operation(operations.Score(3, False)) operations_graph.append_operation(operations.Score(3, False))
keep_best = operations.KeepBestN(3, True) keep_best = operations.KeepBestN(3, True)
operations_graph.append_operation(keep_best) operations_graph.append_operation(keep_best)
@ -557,7 +556,7 @@ def got() -> operations.GraphOfOperations:
keep_best2 = operations.KeepBestN(1, True) keep_best2 = operations.KeepBestN(1, True)
keep_best2.add_predecessor(keep_best) keep_best2.add_predecessor(keep_best)
operations_graph.append_operation(keep_best2) operations_graph.append_operation(keep_best2)
operations_graph.append_operation(operations.Generate(1, branch_factor)) operations_graph.append_operation(operations.Generate(1, 10))
operations_graph.append_operation(operations.Score(3, False)) operations_graph.append_operation(operations.Score(3, False))
keep_best3 = operations.KeepBestN(1, True) keep_best3 = operations.KeepBestN(1, True)
keep_best3.add_predecessor(keep_best2) keep_best3.add_predecessor(keep_best2)

BIN
paper/final_results_gpt35.tar.bz2
View File

Binary file not shown.