transfer-learning-forecasting/python_env.yaml at main · TzorAL/transfer-learning-forecasting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# Python version required to run the project.
python: '3.8.0'
# Dependencies required to build packages. This field is optional.
build_dependencies:
  - pip
# Dependencies required to run the project.
dependencies:
  - mlflow==1.29.0
  - pytz==2021.1
  - datetime
  - pandas==1.4.3
  - numpy==1.22.4
  - matplotlib==3.5.3
  - darts==0.21.0
  - holidays==0.14.2
  - pycountry==22.3.5
  - python-dateutil==2.8.2 # dateutil
  - click==8.1.3
  - pytest-shutil # shutil
  - torch==1.8.1  # +cpu
  - torchaudio==0.8.1
  - torchmetrics==0.8.1
  - torchvision==0.9.1 #+cpu
  - pytorch-lightning==1.6.4
  # pytorch_lightning==1.5.10 # backwards compatibility problem with optuna
  - optuna==2.10.0
  - plotly==5.10.0 # for optuna visualization plots
  - seaborn==0.11.2
  - scipy==1.6.0
  - scikit-learn==1.1.2
  - pandas-profiling==3.2.0
  - pytz==2021.1
  - protobuf==3.19.4
  - statsmodels==0.13.2
  # pickle==4.0
  - torchensemble==0.1.9
  - python-dotenv-0.21.0
  - tornado-6.2
  - boto3==1.26.29 #remote storage of minio