search

brightmart / text_classification

all kinds of text classification models and more with deep learning
MIT License
7.87k stars 2.57k forks source link

does not support multi-label classification #43

Closed acadTags closed 6 years ago

acadTags commented 6 years ago

Thank you for sharing these wonderful codes!

Just one issue, it seems in a03 TextRNN, the result part (self.predictions) does not support multi-label classification. There is a lot to change if I want to adapt it to a multi-label classification task. Also the inference part is mostly about single label classification.

Best wishes, Hang

brightmart commented 6 years ago

UPDATE: if you want to try a model now, you can go to folder 'a02_TextCNN', run 'python -u p7_TextCNN_train.py', it will use sample data to train a model, and print loss and F1 score periodically.