numpy>=1.21
scikit-learn>=0.24.2torch>=1.9.0
