torch>=1.4
numpy
scikit-learn
