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