RandolphVI / Multi-Label-Text-Classification

About Muti-Label Text Classification Based on Neural Network.
Apache License 2.0
553 stars 143 forks source link

How to solve the problem that topK's K is different for every input text? #16

Closed guotong1988 closed 4 years ago

guotong1988 commented 4 years ago

The output topK's K is fixed now.

Do you think training a classifier to predict the value of K for every input is a good solution?

Thank you very much.

RandolphVI commented 4 years ago

@guotong1988

First, how to solve the problem that K of topK is different for every input text?

Second, training a classifier to predict the value of K for every input is a good solution?

Hope it helps!

guotong1988 commented 4 years ago

Thank you. In summary, threshold is one solution. The solution of mine is ok. Am I right?

RandolphVI commented 4 years ago

@guotong1988 Generally speaking, yes.