Decision Tree in Python : Heart Failure Prediction

Python code

R code

Data Analysis Results

---

Decision Tree

We have implemented the following criteria to split nodes in our Decision Trees for comparison of accuracy

• Gini Index
• Information Gain (Entropy)

We have taken (2/3)rd of our Heart Failure Prediction Dataset to be used as Training Dataset & (1/3)rd to be used as Testing Dataset

---

Gini Index

Confusion Matrix: Prediction on Test Dataset

Accuracy = 72/97 = 74.23 %

---

Information Gain (Entropy)

Confusion Matrix: Prediction on Test Dataset

Accuracy = 72/97 = 80.41 %

---