torch==1.9.0
numpy==1.17.4
pandas==0.25.3
scikit_learn==0.22.1
