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

Make target label work with multi class. #10

Open
sgoede opened this issue Oct 12, 2021 · 1 comment
Open

Make target label work with multi class. #10

sgoede opened this issue Oct 12, 2021 · 1 comment

Comments

@sgoede
Copy link

sgoede commented Oct 12, 2021

As target_label only accept single values, making it work on multi class problems is not so simple. I have written a simple looping function to iterate over the correct target labels, but as on Flair 0.8post1, I find that predictions of the interpret_sentence class deviate from the orginal multi class model (i.e. wrong class prediction).

On personal note: I think you have done a really good job in building this solution. It has worked wonders for multi label problems already.

@robinvanschaik
Copy link
Owner

Hi @sgoede,

How nice to hear. Thanks for adding a comment to illustrate this issue.

Are you willing to contribute your solution with a PR? :)

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

2 participants