torch==1.10.0
matplotlib
numpy
tqdm
scikit-learn
hyperopt
