Predicting employee churn using decision trees

HR Analytics: Predicting Employee Churn in Python

Hrant Davtyan

Assistant Professor of Data Science American University of Armenia

Decision Tree in Python

from sklearn.tree import DecisionTreeClassifier

model = DecisionTreeClassifier(random_state=42)

model.fit(features_train,target_train)

model.score(features_test,target_test)*100

Let's practice!

HR Analytics: Predicting Employee Churn in Python