pip install torch
pip install sklearn