jiegzhan/multi-class-text-classification-cnn

Changing Labels

johnp2266 opened this issue · 2 comments

johnp2266 commented

If I want to change the labels, say I want to classify 7 labels, do I just change labels.json?

jiegzhan commented

labels are collected from training data automatically.

multi-class-text-classification-cnn/train.py

Line 39 in 8961f92

"""Step 4: save the labels into labels.json since predict.py needs it"""

johnp2266 commented

Side question, why the accuracy so low? I tried training with the original files but instead of 1 epoch, I changed to 10. That's all I changed but the accuracy is only 89 when I predict.