numpy
matplotlib
torch==1.13.1 
scikit-learn
argparse
pandas