matplotlib==3.1.1 numpy==1.17.4 pandas==0.25.3 scipy==1.4.1 torch==1.2.0 scikit_learn==0.23.1