Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

function for confusion matrix #1264

Open
mmastand opened this issue Sep 25, 2018 · 1 comment
Open

function for confusion matrix #1264

mmastand opened this issue Sep 25, 2018 · 1 comment

Comments

@mmastand
Copy link
Contributor

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 mmastand self-assigned this Sep 28, 2018
@mmastand mmastand changed the title evaluate.multiclass should print a confusion matrix. function for confusion matrix Sep 28, 2018
@mmastand
Copy link
Contributor Author

mmastand commented Sep 28, 2018

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.

  • caret::confusionMatrix has a nice print method but doesn't return a helpful table.
  • The code in plot.multiclass_predicted_df makes a nice tibble but would need its own print method.

Improvement from writing our own might be marginal.

@mmastand mmastand removed their assignment Sep 28, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant