torch>=2.0
scikit-learn>=1.2
