From c6cc8a30e12886c86e16caf005c564f639556aca Mon Sep 17 00:00:00 2001 From: Young Date: Tue, 9 Jul 2024 03:42:40 +0000 Subject: [PATCH] Scaffold --- .../{data mining => data_mining}/README.md | 0 rdagent/app/data_mining/run.py | 28 +++++++++++++++++++ rdagent/scenarios/data_mining/__init__.py | 14 ++++++++++ 3 files changed, 42 insertions(+) rename rdagent/app/{data mining => data_mining}/README.md (100%) create mode 100644 rdagent/app/data_mining/run.py create mode 100644 rdagent/scenarios/data_mining/__init__.py diff --git a/rdagent/app/data mining/README.md b/rdagent/app/data_mining/README.md similarity index 100% rename from rdagent/app/data mining/README.md rename to rdagent/app/data_mining/README.md diff --git a/rdagent/app/data_mining/run.py b/rdagent/app/data_mining/run.py new file mode 100644 index 000000000..2be32b08a --- /dev/null +++ b/rdagent/app/data_mining/run.py @@ -0,0 +1,28 @@ + + + +""" +Task +""" + +# rdagent/app/qlib_rd_loop/factor.py : workflow +# you can use clas to directly pass the content to next step +# rdagent/components/coder/factor_coder/factor.py + + +# TODO: define task +# TODO: define implementation +# - example rdagent/components/coder/model_coder/model.py +# - use `DockerEnv` to run the folder +# TODO: Define your environment +imp = task_gen.gen(task) +# For implementation: +# - rdagent/components/coder/model_coder/one_shot/__init__.py +imp.execute() + +# TODO: +# - Summarize to things that the large language can under stand. + + + +# - TODO: some steps are not used. diff --git a/rdagent/scenarios/data_mining/__init__.py b/rdagent/scenarios/data_mining/__init__.py new file mode 100644 index 000000000..6045242f6 --- /dev/null +++ b/rdagent/scenarios/data_mining/__init__.py @@ -0,0 +1,14 @@ + + + +from rdagent.core.experiment import FBImplementation + + +class DMImplementation(FBImplementation): + # inject code: + def prepare(self, *args, **kwargs): + # create the folder and input necessary dependencies + ... + + def execute(self, *args, **kwargs): + # run code & data in the folder