numpy>=1.20
matplotlib>=3.0
scikit-learn>=0.24
torch>=1.8
