Skip to content
This repository has been archived by the owner on Jul 18, 2024. It is now read-only.

Does it support for multi-label classification #2

Open
xdwang0726 opened this issue Aug 8, 2019 · 1 comment
Open

Does it support for multi-label classification #2

xdwang0726 opened this issue Aug 8, 2019 · 1 comment

Comments

@xdwang0726
Copy link

Hi, I just wondering does it support for multi-label classification as in the example, the pre-processed dataset seems to fit for a binary classification task. I am wondering, if I want to use the code for multi-label classification, how could I deal with the "doc1_label_ID" in the prepared text file. Thank you in advance for your help!

@teddylfwu
Copy link
Contributor

Yes, it is able to support multi-label classification with very minor modification in the training classifier. The WME is essentially providing only the embeddings of each sentence/document and can be combined with any classifier/clustering task.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants