torch>=1.9.0
scikit-learn>=1.0.0
