mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-05 11:07:43 +00:00
Xu Yang
f2745c3cc0
* use CoSTEER as component name * rename factorimplementation to avoid confusion * rename modelimplementation * align benchmark and evolving evaluators * add scenario to evaluator init function * rename all factorimplementationknowledge in CoSTEER * remove all scenario related information in component * remove useless code --------- Co-authored-by: xuyang1 <xuyang1@microsoft.com>
17 lines
497 B
Python
17 lines
497 B
Python
from rdagent.components.coder.model_coder.model import ModelImplementation
|
|
from rdagent.core.task_generator import TaskGenerator
|
|
|
|
|
|
class QlibModelImplementation(TaskGenerator[ModelImplementation]):
|
|
"""
|
|
Docker run
|
|
Everything in a folder
|
|
- config.yaml
|
|
- Pytorch `model.py`
|
|
- results in `mlflow`
|
|
|
|
https://github.com/microsoft/qlib/blob/main/qlib/contrib/model/pytorch_nn.py
|
|
- pt_model_uri: hard-code `model.py:Net` in the config
|
|
- let LLM modify model.py
|
|
"""
|