torch
numpy
python-dotenv
scikit_learn
