numpy
scikit-learn
torch>=1.10.0
pandas
pkbar
matplotlib
scipy
