tensorflow-datasets>=4.9.3
torch >= 1.3
datasets
scikit-learn
tensorflow