Inspiration

As a lot of people are suffering from late diagnosis of heart diseases we wanted to do our part and hence made a program with only a few inputs from the user can predict weather the patient is in risk of a heart disease or not

Working

We used a Random forest separation model and the datasets are taken from Kaggle and It uses the train_test_split function from scikit-learn to split the data into training and testing sets. 80% of the data is used for training (assigned to X_train and y_train), and 20% is used for testing (assigned to X_test and y_test). The code trains the Random Forest Classifier model (model) on the training data using model.fit(X_train, y_train).

Challenges we ran into

The challenges we ran into was finding all the different data and understanding what it means and how these different data can be related to heart problems and how to train the ML model to use these different data sets .As engineers these medical data was a challenge for us to process and use.

Accomplishments that we're proud of

We are proud that we made a working model with nearly 90% accuracy

What we learned

We learnt a lot about the causes of heart diseases and how to control them is in our hands and on how we could leverage the IBM servers and use LinuxOne and Jupyter as efficient tools to run ML models and learnt on how to search,create and use various data sets

What's next for Heart Disease predictor

As the model and data sets increases we can deploy this in hospitals to make heart disease prediction easier and also make it a open-source software as anyone in any part of the world can access this model and self evaluate themselves and see weather they have a heart disease or not

Built With

  • jupyter
  • kaggle
Share this project:

Updates