-
-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
F1, precision & recall for multi-class classification #772
Comments
Sure, we will add support for multi class. BTW, |
It is added to master branch now. See tests for examples. |
Hi @haifengl, Thanks for your pointer to Thanks! |
Hi @haifengl, Could you make a new release? Sander |
The new feature has to change the class APIs so that we couldn't include it in a bug release such as 3.1.1. But it will be available in next major release, which we expect soon. Meanwhile, please test it with master branch and let's know if need any additional work. Thanks! |
Is your feature request related to a problem? Please describe.
I am evaluating a multi-class classification case. The values in the target column and prediction columns can be 0, 1, 2, 3 or 4.
The FScore, Precision and Recall calculation results in the following error:
Describe the solution you'd like
Could you implement this, please? Calculating precision and recall for multi-class classification should be possible.
Describe alternatives you've considered
Not applicable.
Additional context
Not applicable.
The text was updated successfully, but these errors were encountered: