1. role.py - 如果已经有plan,便不再重复生成

2. 修改prompt,让predictions.csv生成的格式与原gt格式一样
This commit is contained in:
Yizhou Chi 2024-09-02 15:15:53 +08:00
parent 0e5db1c364
commit f28908e4c5
5 changed files with 14 additions and 11 deletions

View file

@ -46,7 +46,7 @@ class AugExperimenter(Experimenter):
"score_dict": score_dict,
"aug_mode": self.args.aug_mode,
"insights" : exps[i],
"user_requirement": user_requirement,
"user_requirement": requirement,
"args": vars(self.args)
})
scores = [result["score_dict"]["test_score"] for result in results]

View file

@ -25,9 +25,9 @@ class MCTSExperimenter(Experimenter):
self.save_tree(text)
results = {
"best_node": best_node,
"best_node": best_node.id,
"best_node_score": best_node.raw_reward,
"dev_best_node": dev_best_node,
"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"],