scikit-learn==1.5.1
torchmetrics==1.4.0