HealthCatalyst / healthcareai-r

R tools for healthcare machine learning
https://docs.healthcare.ai
Other
245 stars 106 forks source link

function for confusion matrix #1264

Open mmastand opened 6 years ago

mmastand commented 6 years ago

Create an evaluate method multiclass so that a confusion matrix is printed as a tibble. I think most of the code to do this is already in plot.multiclass_multiclass_df

mmastand commented 6 years ago

I had originally thought this might work well under evaluate.multiclass, but now I don't think so. That function is for returning performance stats of best or all models. I think the correct way to implement this would be as a separate user-facing function that would get a confusion matrix for both binary and multiclass. I'm not sure where this would go in the package or how a user would know to use it.

Improvement from writing our own might be marginal.