torch
torchvision
sklearn