A basic neural network in pytorch used to predict heart disease based on various parameters from a patient.
The neural network receives 13 parameters and outputs a value between 0 and 1 as a probability for a heart disease.
Dataset source: https://www.kaggle.com/datasets/johnsmith88/heart-disease-dataset/data