attrition_confusion_maxtrix.py

print("TRAIN PERFORMANCE:\n")
confusion_matrix_train = confusion_matrix(y_train, model.predict(X_train))
confusion_matrix_train = pd.DataFrame(confusion_matrix_train,
                                      index=["Actual_No","Actual_Yes"],
                                      columns=["Predicted_No","Predicted_Yes"])

display(confusion_matrix_train)

recall_resignation_train = confusion_matrix_train.iloc[1,1] / confusion_matrix_train.iloc[1,:].sum()
print("Train Score: {}".format(round(model.score(X_train,y_train),3)))
print("Recall Train: {}".format(round(recall_resignation_train,3)))
print("\n* * * * * * * * * * * * * * * * * * *\n")
print("TEST PERFORMANCE:\n")
confusion_matrix_test = confusion_matrix(y_test, model.predict(X_test))
confusion_matrix_test = pd.DataFrame(confusion_matrix_test,
                                     index=["Actual_No","Actual_Yes"],
                                     columns=["Predicted_No","Predicted_Yes"])

display(confusion_matrix_test)

recall_resignation_test = confusion_matrix_test.iloc[1,1] / confusion_matrix_test.iloc[1,:].sum()
print("Train Score: {}".format(round(model.score(X_test,y_test),3)))
print("Recall Train: {}".format(round(recall_resignation_test,3)))