numpy==1.19.5
scikit-learn==1.0.2
torch==1.10.0
torchvision==0.11.1
matplotlib
seaborn==0.11.2
Pillow
