From e10e7f0e597563c2ebb682a3bd02aea009e208c5 Mon Sep 17 00:00:00 2001 From: Suhan Cui <51844791+SH-Src@users.noreply.github.com> Date: Thu, 25 Jul 2024 16:12:52 +0800 Subject: [PATCH] fix bugs (#111) --- rdagent/scenarios/data_mining/experiment/model_experiment.py | 4 ++++ rdagent/scenarios/data_mining/proposal/model_proposal.py | 4 +++- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/rdagent/scenarios/data_mining/experiment/model_experiment.py b/rdagent/scenarios/data_mining/experiment/model_experiment.py index ff90bf8f..f737f83d 100644 --- a/rdagent/scenarios/data_mining/experiment/model_experiment.py +++ b/rdagent/scenarios/data_mining/experiment/model_experiment.py @@ -38,6 +38,10 @@ class DMModelScenario(Scenario): @property def simulator(self) -> str: return prompt_dict["dm_model_simulator"] + + @property + def rich_style_description(self)->str: + return "Below is MIMIC Model Evolving Automatic R&D Demo." def get_scenario_all_desc(self) -> str: return f"""Background of the scenario: diff --git a/rdagent/scenarios/data_mining/proposal/model_proposal.py b/rdagent/scenarios/data_mining/proposal/model_proposal.py index 2f420944..32ec8896 100644 --- a/rdagent/scenarios/data_mining/proposal/model_proposal.py +++ b/rdagent/scenarios/data_mining/proposal/model_proposal.py @@ -84,10 +84,12 @@ class DMModelHypothesis2Experiment(ModelHypothesis2Experiment): tasks = [] for model_name in response_dict: description = response_dict[model_name]["description"] + formulation = response_dict[model_name]["formulation"] architecture = response_dict[model_name]["architecture"] + variables = response_dict[model_name]["variables"] hyperparameters = response_dict[model_name]["hyperparameters"] model_type = response_dict[model_name]["model_type"] - tasks.append(ModelTask(model_name, description, architecture, hyperparameters, model_type)) + tasks.append(ModelTask(model_name, description, formulation, architecture, variables, hyperparameters, model_type)) exp = DMModelExperiment(tasks) exp.based_experiments = [t[1] for t in trace.hist if t[2]] return exp