├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md └── workflows │ ├── auto-make.yml │ └── auto-unit-test.yml ├── .gitignore ├── CHANGELOG.md ├── CODE_OF_CONDUCT.md ├── Jenkinsfile ├── LICENSE ├── Makefile ├── README.md ├── code-env └── python │ ├── desc.json │ └── spec │ └── requirements.txt ├── custom-recipes ├── timeseries-forecast-1-train-evaluate │ ├── recipe.json │ └── recipe.py └── timeseries-forecast-2-predict │ ├── recipe.json │ └── recipe.py ├── plugin.json ├── python-lib ├── __init__.py ├── dku_config │ ├── README.md │ ├── __init__.py │ ├── custom_check.py │ ├── dku_config.py │ ├── dss_parameter.py │ └── requirements.txt ├── dku_constants.py ├── dku_io_utils │ ├── checks_utils.py │ ├── config_handler.py │ ├── dku_file_manager.py │ ├── model_selection.py │ ├── partitions_handling.py │ └── utils.py ├── gluonts_forecasts │ ├── custom_models │ │ ├── autoarima.py │ │ ├── seasonal_trend.py │ │ └── utils.py │ ├── gluon_dataset.py │ ├── model.py │ ├── model_config.py │ ├── model_config_registry.py │ ├── mxnet_utils.py │ ├── trained_model.py │ ├── training_session.py │ └── utils.py ├── safe_logger.py └── timeseries_preparation │ ├── h2g2.py │ └── preparation.py ├── resource ├── select_gpu_devices.py └── select_model_version.py └── tests └── python ├── integration ├── pytest.ini ├── requirements.txt └── test_scenario.py └── unit ├── requirements.txt ├── test_autoarima.py ├── test_cross_validation.py ├── test_frequencies.py ├── test_gluon_dataset.py ├── test_gluonts_forecasts_utils.py ├── test_missing_values.py ├── test_model.py ├── test_seasonal_trend.py ├── test_timeseries_preparation.py ├── test_trained_model.py └── test_training_session.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/workflows/auto-make.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/.github/workflows/auto-make.yml -------------------------------------------------------------------------------- /.github/workflows/auto-unit-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/.github/workflows/auto-unit-test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /Jenkinsfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/Jenkinsfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/README.md -------------------------------------------------------------------------------- /code-env/python/desc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/code-env/python/desc.json -------------------------------------------------------------------------------- /code-env/python/spec/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/code-env/python/spec/requirements.txt -------------------------------------------------------------------------------- /custom-recipes/timeseries-forecast-1-train-evaluate/recipe.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/custom-recipes/timeseries-forecast-1-train-evaluate/recipe.json -------------------------------------------------------------------------------- /custom-recipes/timeseries-forecast-1-train-evaluate/recipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/custom-recipes/timeseries-forecast-1-train-evaluate/recipe.py -------------------------------------------------------------------------------- /custom-recipes/timeseries-forecast-2-predict/recipe.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/custom-recipes/timeseries-forecast-2-predict/recipe.json -------------------------------------------------------------------------------- /custom-recipes/timeseries-forecast-2-predict/recipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/custom-recipes/timeseries-forecast-2-predict/recipe.py -------------------------------------------------------------------------------- /plugin.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/plugin.json -------------------------------------------------------------------------------- /python-lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python-lib/dku_config/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_config/README.md -------------------------------------------------------------------------------- /python-lib/dku_config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_config/__init__.py -------------------------------------------------------------------------------- /python-lib/dku_config/custom_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_config/custom_check.py -------------------------------------------------------------------------------- /python-lib/dku_config/dku_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_config/dku_config.py -------------------------------------------------------------------------------- /python-lib/dku_config/dss_parameter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_config/dss_parameter.py -------------------------------------------------------------------------------- /python-lib/dku_config/requirements.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /python-lib/dku_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_constants.py -------------------------------------------------------------------------------- /python-lib/dku_io_utils/checks_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_io_utils/checks_utils.py -------------------------------------------------------------------------------- /python-lib/dku_io_utils/config_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_io_utils/config_handler.py -------------------------------------------------------------------------------- /python-lib/dku_io_utils/dku_file_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_io_utils/dku_file_manager.py -------------------------------------------------------------------------------- /python-lib/dku_io_utils/model_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_io_utils/model_selection.py -------------------------------------------------------------------------------- /python-lib/dku_io_utils/partitions_handling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_io_utils/partitions_handling.py -------------------------------------------------------------------------------- /python-lib/dku_io_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/dku_io_utils/utils.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/custom_models/autoarima.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/custom_models/autoarima.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/custom_models/seasonal_trend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/custom_models/seasonal_trend.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/custom_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/custom_models/utils.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/gluon_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/gluon_dataset.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/model.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/model_config.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/model_config_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/model_config_registry.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/mxnet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/mxnet_utils.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/trained_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/trained_model.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/training_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/training_session.py -------------------------------------------------------------------------------- /python-lib/gluonts_forecasts/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/gluonts_forecasts/utils.py -------------------------------------------------------------------------------- /python-lib/safe_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/safe_logger.py -------------------------------------------------------------------------------- /python-lib/timeseries_preparation/h2g2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/timeseries_preparation/h2g2.py -------------------------------------------------------------------------------- /python-lib/timeseries_preparation/preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/python-lib/timeseries_preparation/preparation.py -------------------------------------------------------------------------------- /resource/select_gpu_devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/resource/select_gpu_devices.py -------------------------------------------------------------------------------- /resource/select_model_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/resource/select_model_version.py -------------------------------------------------------------------------------- /tests/python/integration/pytest.ini: -------------------------------------------------------------------------------- 1 | [pytest] 2 | usefixtures = plugin dss_target 3 | -------------------------------------------------------------------------------- /tests/python/integration/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/integration/requirements.txt -------------------------------------------------------------------------------- /tests/python/integration/test_scenario.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/integration/test_scenario.py -------------------------------------------------------------------------------- /tests/python/unit/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/requirements.txt -------------------------------------------------------------------------------- /tests/python/unit/test_autoarima.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_autoarima.py -------------------------------------------------------------------------------- /tests/python/unit/test_cross_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_cross_validation.py -------------------------------------------------------------------------------- /tests/python/unit/test_frequencies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_frequencies.py -------------------------------------------------------------------------------- /tests/python/unit/test_gluon_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_gluon_dataset.py -------------------------------------------------------------------------------- /tests/python/unit/test_gluonts_forecasts_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_gluonts_forecasts_utils.py -------------------------------------------------------------------------------- /tests/python/unit/test_missing_values.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_missing_values.py -------------------------------------------------------------------------------- /tests/python/unit/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_model.py -------------------------------------------------------------------------------- /tests/python/unit/test_seasonal_trend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_seasonal_trend.py -------------------------------------------------------------------------------- /tests/python/unit/test_timeseries_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_timeseries_preparation.py -------------------------------------------------------------------------------- /tests/python/unit/test_trained_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_trained_model.py -------------------------------------------------------------------------------- /tests/python/unit/test_training_session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dataiku/dss-plugin-timeseries-forecast/HEAD/tests/python/unit/test_training_session.py --------------------------------------------------------------------------------