Predikce odchodu zaměstnanců pomocí rozhodovacích stromů

HR Analytics: Predicting Employee Churn in Python

Hrant Davtyan

Assistant Professor of Data Science American University of Armenia

Rozhodovací strom v Pythonu

from sklearn.tree import DecisionTreeClassifier

model = DecisionTreeClassifier(random_state=42)

model.fit(features_train,target_train)

model.score(features_test,target_test)*100

Pojďme si procvičit!

HR Analytics: Predicting Employee Churn in Python