numpy>=1.19.1
matplotlib>=3.3.0
scikit-learn>=0.23.2
torch>=1.6.0+cpu
