hellonlp / classifier_multi_label

multi-label，classifier，text classification，多标签文本分类，文本分类，BERT，ALBERT，multi-label-classification

简介

1、本项目是在tensorflow版本1.15.0的基础上做的训练和测试。
2、本项目为中文的多标签文本分类。
3、欢迎大家联系我 www.hellonlp.com
4、albert_small_zh_google对应的百度云下载地址：
链接：https://pan.baidu.com/s/1RKzGJTazlZ7y12YRbAWvyA
提取码：wuxw

使用方法

1、准备数据
数据格式为：classifier_multi_label/data/test_onehot.csv
2、参数设置
参考脚本 hyperparameters.py，直接修改里面的数值即可。
3、训练
python train.py
4、预测
python predict.py

知乎代码解读

https://zhuanlan.zhihu.com/p/164873441

classifier_multi_label's Issues

请问楼主数据格式代码有写么？

请问楼主数据格式代码有写么？

Unsuccessful TensorSliceReader constructor

Failed to find any matching files for C:\Users\Administrator\D
esktop\classifier_multi_label\albert_small_zh_google\albert_model.ckpt
请问需要提前准备albert_model.ckpt吗

Assign requires shapes of both tensors to match. lhs shape= [22,384] rhs shape= [15,384]

请问为了解决这个报错，shape应该在哪里修改？

No module named 'classifier_multi_label' 这是什么原因呢

tensorflow1.x版本最高就1.14.0

安装tensorflow==1.14.0后，代码报错没有tensorflow-hub，安装后又要安装tensorflow==1.15.0，问题是tensorflow最高版本是1.14.0，这该怎么解决呢？

vocab list

Can you share an English vocabulary txt? I want to test the model on reviews. Thanks.
（albert_small_zh_google dir. contains vocab_chinese.txt）

networks.py问题

这里logits是不是应该为self.probabilities