【文件属性】:
文件名称:classifier_multi_label:multi-label,classifier,text classification,多标签文本分类,文本分类,BERT,ALBERT,multi-label-classification
文件大小:490KB
文件格式:ZIP
更新时间:2021-05-03 21:59:21
text-classification tensorflow multi-label-classification albert bert
简介
1、本项目是在tensorflow版本1.14.0的基础上做的训练和测试。
2、本项目为中文的多标签文本分类。
3、欢迎大家联系我
4、albert_small_zh_google对应的百度云下载地址:
链接:
提取码:wuxw
使用方法
1、准备数据
数据格式为:classifier_multi_label/data/test_onehot.csv
2、参数设置
参考脚本 hyperparameters.py,直接修改里面的数值即可。
3、训练
python train.py
4、预测
python predict.py
知乎代码解读
【文件预览】:
classifier_multi_label-master
----albert_small_zh_google()
--------vocab_chinese.txt(107KB)
--------albert_config.json(482B)
----tokenization.py(16KB)
----data()
--------test_onehot.csv(9KB)
--------test.csv(5KB)
--------train.csv(63KB)
--------vocabulary_label.txt(160B)
--------train_onehot.csv(116KB)
----model()
--------model_01()
--------saved_01()
----classifier_utils.py(27KB)
----logdir()
--------model_01()
----__init__.py(606B)
----optimization.py(7KB)
----modeling.py(42KB)
----utils.py(2KB)
----README.md(660B)
----image()
--------graph.png(117KB)
----hyperparameters.py(2KB)
----lamb_optimizer.py(5KB)
----networks.py(6KB)
----modules.py(2KB)
----predict.py(2KB)
----train.py(3KB)
网友评论
相关文章
- classifier_multi_label_textcnn:multi-label,classifier,text classification,多标签文本分类,文本分类,BERT,ALBERT,multi-label-classification
- sentiment_analysis_albert:sentiment analysis、文本分类、ALBERT、TextCNN、classification、tensorflow、BERT、CNN、text classification
- Multi-label-classification:多标签分类
- Text-Multilabel-Classification-BERT-Tf2.x:文本多标签分类-BERT-Tf2.0