numpy==1.21.0 pandas==1.1.2 scikit_learn==0.23.2 torch==1.7.0