Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

updated weighted_f1 to not assume binary classification #3728

Merged
merged 1 commit into from
Jun 16, 2021

Conversation

arendu-zz
Copy link
Contributor

Patch description
weighted f1 was calculated was broken -- it assumes there are only 2 classes. Specifically total_examples calculation was corrected for multiclass.

Testing steps

Other information

Copy link
Contributor

@emilydinan emilydinan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thank you for the fix!

@arendu-zz arendu-zz merged commit 76b8804 into master Jun 16, 2021
@arendu-zz arendu-zz deleted the fix_multiclass_weighted_f1 branch June 16, 2021 21:04
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants