mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-27 23:47:46 +00:00
c707a40073
* scen * scen2 * app * fix * Simplify workflow * We can share more code in new scenarios * rename model to rd loop * Optimize data path * Update rdagent/app/data_mining/model.py * Add TODO * Support GPU * gpu --------- Co-authored-by: SH-Src <suhan.c@outlook.com>
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from pathlib import Path
|
|
|
|
|
|
from rdagent.components.workflow.conf import BasePropSetting
|
|
|
|
|
|
class PropSetting(BasePropSetting):
|
|
class Config:
|
|
env_prefix = "DM_" # Use MODEL_CODER_ as prefix for environment variables
|
|
protected_namespaces = () # Add 'model_' to the protected namespaces
|
|
|
|
# 1) overriding the default
|
|
scen: str = "rdagent.scenarios.data_mining.experiment.model_experiment.DMModelScenario"
|
|
hypothesis_gen: str = "rdagent.scenarios.data_mining.proposal.model_proposal.DMModelHypothesisGen"
|
|
hypothesis2experiment: str = "rdagent.scenarios.data_mining.proposal.model_proposal.DMModelHypothesis2Experiment"
|
|
coder: str = "rdagent.scenarios.data_mining.developer.model_coder.DMModelCoSTEER"
|
|
runner: str = "rdagent.scenarios.data_mining.developer.model_runner.DMModelRunner"
|
|
summarizer: str = "rdagent.scenarios.data_mining.developer.feedback.DMModelHypothesisExperiment2Feedback"
|
|
|
|
evolving_n: int = 10
|
|
|
|
# 2) Extra config for the scenario
|
|
# physionet account
|
|
# NOTE: You should apply the account in https://physionet.org/
|
|
username: str = ''
|
|
password: str = ''
|
|
|
|
|
|
PROP_SETTING = PropSetting()
|