Pytorch Machine Learning Models For Classification