jiangxinyang227 / textClassifier

tensorflow implementation
1.14k stars 556 forks source link

bert_blstm_atten.py无法进行多分类 #17

Open zysNLP opened 4 years ago

zysNLP commented 4 years ago

你好,我在修改了bert_blstm_atten.py后发现无法进行多分类验证,训练可以训练,在bert里修改run_classifier就可以。我怀疑是blstm_atten.py这一层只能进行二分类,请问怎么修改代码可以使用bert_blstm进行多分类?