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

add random tree search

Browse source
This commit is contained in:
Yizhou Chi 2024-09-10 15:30:23 +08:00
parent d34a482faf
commit b776c7309b
4 changed files with 23 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
expo/experimenter/mcts.py
View file

@ -1,19 +1,21 @@
from expo.evaluation.visualize_mcts import get_tree_text
from expo.experimenter.experimenter import Experimenter
from expo.Greedy import Greedy
from expo.Greedy import Greedy, Random
from expo.MCTS import MCTS
class MCTSExperimenter(Experimenter):
result_path: str = "results/mcts"
def __init__(self, args, greedy=False, **kwargs):
def __init__(self, args, tree_mode=None, **kwargs):
super().__init__(args, **kwargs)
self.greedy = greedy
self.tree_mode = tree_mode
async def run_experiment(self):
if self.greedy:
if self.tree_mode == "greedy":
mcts = Greedy(root_node=None, max_depth=5)
elif self.tree_mode == "random":
mcts = Random(root_node=None, max_depth=5)
else:
mcts = MCTS(root_node=None, max_depth=5)
best_nodes = await mcts.search(