torch>=1.7
scikit-learn>=1.0
