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

automatically change low_is_better for rmse

Browse source
This commit is contained in:
Yizhou Chi 2024-10-17 17:49:31 +08:00
parent 852fbc58ee
commit 6f437bb76d
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
expo/MCTS.py
View file

@ -43,6 +43,8 @@ def create_initial_state(task, start_task_id, data_config, args):
task = get_mle_task_id(args.custom_dataset_dir)
else:
dataset_config = data_config["datasets"][task]
if dataset_config["metric"] == "rmse":
args.low_is_better = True
datasets_dir = get_split_dataset_path(task, data_config)
requirement = generate_task_requirement(
task, data_config, is_di=True, special_instruction=args.special_instruction