Predicting the death of each patient using Scikit-Learn

Hedris Pereira
5 min readJan 12, 2024

Project developed with Python and Machine Learning using the Scikit-Learn framework, you must also have knowledge of Pandas for manipulating arrays, dataframes and datasets

Introduction

In this article, we will design and develop an application involving patients with heart problems. With this, we will use Scikit-Learn, which is a framework used for machine learning. We will really understand what Scikit-Learn is for beginners, and for developers or More advanced data scientists, we will select the best Scikit-Learn model and test each one.

Dataset reference

The data is from an article by David Chico and Guispe Jurman. Machine learning can predict the survival of patients with heart failure from serum creatinine and ejection fraction alone. BMC Medical Informatics and Decision-Making 20, 16 (2020). Some considerations were made in order to avoid some typical conceptual errors for a beginner on the subject. But if they are not treated now, they could be a big problem in the future.

Ok, let’s actually analyze what Scikit-Learn is.

Scikit-learn is an open-source Python library designed to support machine learning-related tasks. It provides a variety of tools and algorithms…

--

--