Decision Tree in Python : Metaprotein

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)

Metaprotein Dataset

Metaproteins as rows and, patient data of three types with samples from each being tested for the presence of metaproteins in columns (along with metaprotein demographics)

To suit our decision tree model, we removed the demographic columns from the dataset and have transposed the data frame to turn metaproteins into columns/variables & patients as rows.

We created a class label "Patient type" which has 3 factors - C, UC & CD

Metaprotein Dataset after pre-processing

We have taken half (1/2) of our Metaprotein Dataset to be used as Training Dataset & (1/2) to be used as Testing Dataset

---

Gini Index

Confusion Matrix: Prediction on Test Dataset

Accuracy = 19/24 = 79.16 %

---

Information Gain (Entropy)

Confusion Matrix: Prediction on Test Dataset

Accuracy = 17/24 = 70.83 %

---