CODEWITHSUNDEEP
pythonmachine-learningscikit-learndecision-tree
L.G.
L.G.

Reputation: 71

Interpreting Decision Tree in Python

I built a Decision Tree in python and I am struggling to interpret it. The tree look like as picture below.

enter image description here

This a Churn model result. I want to know how can I interpret the following:

  1. Number of children at home <=3.5  (Integer)
  2. MaritalStatus_M <= 0.5  (M- Married in here and was a binary. I was expecting either MaritalStatus_M=0 or =1)
  3. Sales Reason_Price<=0.5 (Binary. I was expecting either Sales Reason_Price=0 or =1)
  4. Tenure_Months<= 0.5 (Integer)

Upvotes: 0

Views: 2026

Answers (1)

desertnaut
desertnaut

Reputation: 60390

There are some internal reasons why split decisions in trees prefer a <= relation (rather than equality, even if the latter might seem to make more sense for a human reader), but such representation details should not pose any difficulty in the interpretation, which is what you are asking about here.

So, practically speaking:

  1. children <= 3.5 means less than four children (assuming variable is integer)
  2. MaritalStatus_M <= 0.5 for a binary variable means MaritalStatus_M==0 (i.e. not married)
  3. Sales Reason_Price<=0.5 for a binary variable means Sales Reason_Price==0
  4. Tenure_Months<= 0.5 means Tenure_Months==0 (assuming variable is integer)

Upvotes: 2

Related Questions