https://ideas.creativecasinoties.com/post/building-a-binary-classification-model-in-pytorch-raudub