matplotlib==3.8.0 numpy==1.26.2 scikit_learn==1.3.0 torch==2.1.0