MetaGPT/expo/Greedy.py
Yizhou Chi a6f71f4498 1. avoid circular reference
2. add greedy
2024-09-06 13:09:18 +08:00

9 lines
325 B
Python

from expo.MCTS import MCTS
class Greedy(MCTS):
def best_child(self):
if len(self.children) == 0:
return self.root_node
all_children = [child for children in self.children.values() for child in children]
return max(all_children, key=lambda x: x.normalized_reward.get("dev_score", 0))