https://machinelearningmastery.com/building-a-logistic-regression-classifier-in-pytorch/