funModeling (version 1.0)

model_performance: Get model perfomance metrics (KS, AUC and ROC)

Description

Get model performance for tree models (rpart library), or glm. It returns quality metrics: AUC (Area Under ROC Curve) and KS (Kolmogorov-Smirnov), and plots the ROC curve.

Usage

model_performance(fit, data, target_var)

Value

None.

Examples

fit_glm=glm(has_heart_disease ~ age + oldpeak, data=heart_disease, family = binomial)
model_performance(fit=fit_glm, data = heart_disease, target_var = "has_heart_disease")