feat: add kaggle command (#271)

* feat: add kaggle command

* lint
This commit is contained in:
you-n-g
2024-09-18 16:48:54 +08:00
committed by GitHub
parent 204580bc52
commit 72e7c3b83a
2 changed files with 4 additions and 4 deletions
+2
View File
@@ -20,6 +20,7 @@ from rdagent.app.data_mining.model import main as med_model
from rdagent.app.general_model.general_model import (
extract_models_and_implement as general_model,
)
from rdagent.app.kaggle.loop import main as kaggle_main
from rdagent.app.qlib_rd_loop.factor import main as fin_factor
from rdagent.app.qlib_rd_loop.factor_from_report import main as fin_factor_report
from rdagent.app.qlib_rd_loop.model import main as fin_model
@@ -51,5 +52,6 @@ def app():
"general_model": general_model,
"ui": ui,
"collect_info": collect_info,
"kaggle": kaggle_main,
}
)
+2 -4
View File
@@ -77,9 +77,10 @@ def main(path=None, step_n=None, competition=None):
You can continue running session by
.. code-block:: python
.. code-block:: bash
dotenv run -- python rdagent/app/kaggle/loop.py [--competition titanic] $LOG_PATH/__session__/1/0_propose --step_n 1 # `step_n` is a optional paramter
rdagent kaggle --competition playground-series-s4e8 # You are encouraged to use this one.
"""
if competition:
@@ -92,7 +93,4 @@ def main(path=None, step_n=None, competition=None):
if __name__ == "__main__":
from dotenv import load_dotenv
load_dotenv(override=True)
fire.Fire(main)