interpret-flair icon indicating copy to clipboard operation
interpret-flair copied to clipboard

Make target label work with multi class.

Open sgoede opened this issue 3 years ago • 1 comments

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.

sgoede avatar Oct 12 '21 09:10 sgoede

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? :)

robinvanschaik avatar Oct 12 '21 10:10 robinvanschaik