pytorch_lightning<=1.4.0,>=1.1.3
scipy
scikit-learn
