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

update prompt (specify whether each set has target label)

Browse source
This commit is contained in:
Yizhou Chi 2024-09-03 16:31:03 +08:00
parent df877c973e
commit 6972afb755
5 changed files with 22 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

23
expo/experimenter/mcts.py
View file

@ -22,19 +22,18 @@ class MCTSExperimenter(Experimenter):
text += f"Best node: {best_node}, score: {best_node.raw_reward}\n"
text += f"Dev best node: {dev_best_node}, score: {dev_best_node.raw_reward}\n"
print(text)
if self.args.rollouts > 0:
self.save_tree(text)
self.save_tree(text)
results = {
"best_node": best_node.id,
"best_node_score": best_node.raw_reward,
"dev_best_node": dev_best_node.id,
"dev_best_node_score": dev_best_node.raw_reward,
"num_generated_codes": num_generated_codes,
"user_requirement": best_node.state["requirement"],
"args": vars(self.args)
}
self.save_result(results)
results = {
"best_node": best_node.id,
"best_node_score": best_node.raw_reward,
"dev_best_node": dev_best_node.id,
"dev_best_node_score": dev_best_node.raw_reward,
"num_generated_codes": num_generated_codes,
"user_requirement": best_node.state["requirement"],
"args": vars(self.args)
}
self.save_result(results)