save result order

This commit is contained in:
Yizhou Chi 2024-09-28 15:01:01 +08:00
parent 06702db6d1
commit db247d9ff4

View file

@ -36,8 +36,6 @@ class MCTSExperimenter(Experimenter):
text += f"Best node: {best_node.id}, score: {best_node.raw_reward}\n"
text += f"Dev best node: {dev_best_node.id}, score: {dev_best_node.raw_reward}\n"
print(text)
self.save_tree(text)
results = [
{
"best_node": best_node.id,
@ -54,6 +52,7 @@ class MCTSExperimenter(Experimenter):
self.save_result(results)
self.copy_notebook(best_node, "best")
self.copy_notebook(dev_best_node, "dev_best")
self.save_tree(text)
def copy_notebook(self, node, name):
node_dir = node.get_node_dir()