torch>=2.0.0
tqdm
numpy
matplotlib
scikit-learn
