numpy==1.18.1
scikit-learn==0.22.1
scipy==1.3.2
matplotlib==3.1.2
torch==1.6.0
torchvision==0.7.0
hyperopt==0.2.4