New Structure Demo (#120)

better demo
---------

Co-authored-by: Young <afe.young@gmail.com>
Co-authored-by: Taozhi Wang <taozhi.mark.wang@gmail.com>
Co-authored-by: you-n-g <you-n-g@users.noreply.github.com>
Co-authored-by: cyncyw <47289405+taozhiwang@users.noreply.github.com>
This commit is contained in:
XianBW
2024-07-30 17:23:05 +08:00
committed by GitHub
parent 1b4e98a752
commit dbbec2ffaf
11 changed files with 984 additions and 73 deletions
@@ -1,13 +1,12 @@
from pathlib import Path
from rdagent.components.coder.model_coder.model import (
ModelExperiment,
)
from rdagent.components.coder.model_coder.model import ModelExperiment
from rdagent.core.prompts import Prompts
from rdagent.core.scenario import Scenario
prompt_dict = Prompts(file_path=Path(__file__).parent / "prompts.yaml")
class GeneralModelScenario(Scenario):
@property
def background(self) -> str:
@@ -28,10 +27,10 @@ class GeneralModelScenario(Scenario):
@property
def simulator(self) -> str:
return prompt_dict["general_model_simulator"]
@property
def rich_style_description(self)->str:
return '''
def rich_style_description(self) -> str:
return """
# General Model Scenario
## Overview
@@ -61,7 +60,7 @@ This scenario automates the development of PyTorch models by reading academic pa
- **Time-Series Data:** Sequential data points indexed in time order, useful for forecasting and temporal pattern recognition.
- **Graph Data:** Data structured as nodes and edges, suitable for network analysis and relational tasks.
'''
"""
def get_scenario_all_desc(self) -> str:
return f"""Background of the scenario: