apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-21 14:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add visit order

Browse source
This commit is contained in:
garylin2099 2024-10-21 10:13:36 +00:00
parent d304fc36de
commit 3a8fdc6a5e
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
expo/scripts/visualize_experiment.py
View file

@ -17,7 +17,9 @@ if __name__ == "__main__":
)
mcts.load_tree()
mcts.load_node_order()
root = mcts.root_node
node_order = mcts.node_order
G = nx.DiGraph()
build_tree_recursive(G, "0", root)
build_tree_recursive(G, "0", root, node_order)
visualize_tree(G, save_path=f"results/{args.task}-tree.png")