mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-08 04:27:44 +00:00
Xu Yang
22e1aa3330
* store code into FBImplementation * fix path related bugs * fix a bug * fix factor related small bugs * re-submit all model related code * new code to model coder * finish the model evolving code --------- Co-authored-by: xuyang1 <xuyang1@microsoft.com>
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from pathlib import Path
|
|
|
|
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")
|
|
|
|
QlibModelExperiment = ModelExperiment
|
|
|
|
|
|
class QlibModelScenario(Scenario):
|
|
@property
|
|
def background(self) -> str:
|
|
return prompt_dict["qlib_model_background"]
|
|
|
|
@property
|
|
def source_data(self) -> str:
|
|
raise NotImplementedError("source_data of QlibModelScenario is not implemented")
|
|
|
|
@property
|
|
def output_format(self) -> str:
|
|
return prompt_dict["qlib_model_output_format"]
|
|
|
|
@property
|
|
def interface(self) -> str:
|
|
return prompt_dict["qlib_model_interface"]
|
|
|
|
@property
|
|
def simulator(self) -> str:
|
|
return prompt_dict["qlib_model_simulator"]
|
|
|
|
def get_scenario_all_desc(self) -> str:
|
|
return f"""Background of the scenario:
|
|
{self.background}
|
|
The interface you should follow to write the runnable code:
|
|
{self.interface}
|
|
The output of your code should be in the format:
|
|
{self.output_format}
|
|
The simulator user can use to test your model:
|
|
{self.simulator}
|
|
"""
|