feat: kaggle templates related (#287)

* add kaggle test

* kaggle templates changes
This commit is contained in:
XianBW
2024-09-20 20:49:44 +08:00
committed by GitHub
parent e152c68c88
commit f4dd0ffcd6
19 changed files with 781 additions and 41 deletions
+2
View File
@@ -42,6 +42,8 @@ class KaggleBasePropSetting(BasePropSetting):
competition: str = ""
local_data_path: str = "/data/userdata/share/kaggle"
rag_path: str = "git_ignore_folder/rag"
+5
View File
@@ -18,6 +18,7 @@ from rdagent.core.scenario import Scenario
from rdagent.core.utils import import_class
from rdagent.log import rdagent_logger as logger
from rdagent.log.time import measure_time
from rdagent.scenarios.kaggle.kaggle_crawler import download_data
from rdagent.scenarios.kaggle.proposal.proposal import (
KG_ACTION_FEATURE_ENGINEERING,
KG_ACTION_FEATURE_PROCESSING,
@@ -89,6 +90,10 @@ def main(path=None, step_n=None, competition=None):
"""
if competition:
KAGGLE_IMPLEMENT_SETTING.competition = competition
download_data(competition=competition, local_path=KAGGLE_IMPLEMENT_SETTING.local_data_path)
else:
logger.error("Please specify competition name.")
if path is None:
kaggle_loop = KaggleRDLoop(KAGGLE_IMPLEMENT_SETTING)
else: