mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-02 09:57:44 +00:00
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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user