torchmetrics # necessary because old PL versions don't have it as dependency scikit-learn