diff --git a/dags/config.py b/dags/config.py index 07fbc05..7bbe5e0 100644 --- a/dags/config.py +++ b/dags/config.py @@ -6,9 +6,9 @@ GENERATED_DATA_FOLDER = os.path.join(DATA_FOLDER, "generated_data") PREDICTIONS_FOLDER = os.path.join(DATA_FOLDER, 'predictions') -TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2013-2016.csv') +TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2017-2020.csv') # TODO change, should point to generator -GENERATED_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2013-2016.csv') +GENERATED_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2017-2020.csv') FEATURES_PATH = os.path.join(DATA_FOLDER, 'prepared_features.parquet') diff --git a/scripts/config.py b/scripts/config.py index f615e83..1fcee15 100644 --- a/scripts/config.py +++ b/scripts/config.py @@ -5,9 +5,9 @@ DATA_FOLDER = os.path.join(PROJECT_FOLDER, 'data') PREDICTIONS_FOLDER = os.path.join(DATA_FOLDER, 'predictions') -TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2013-2016.csv') +TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2017-2020.csv') GENERATED_DATA_PATH = os.path.join(DATA_FOLDER, - 'la-haute-borne-data-2013-2016.csv') # TODO change, should point to generator + 'la-haute-borne-data-2017-2020.csv') # TODO change, should point to generator FEATURES_PATH = os.path.join(DATA_FOLDER, 'prepared_features.parquet')