gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-27 23:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bugs (#111)

Browse Source
This commit is contained in:
Suhan Cui
2024-07-25 16:12:52 +08:00
committed by Young
parent 055bffcbf3
commit e10e7f0e59
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/scenarios/data_mining/experiment/model_experiment.py
+4
View File
@@ -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:
rdagent/scenarios/data_mining/proposal/model_proposal.py
+3 -1
View File
@@ -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