2 Star 5 Fork 2

DealiAxy / bert-chinese-classifier

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

bert-chinese-classifier

预训练模型:

https://storage.googleapis.com/bert_models/2018_11_03/chinese_L-12_H-768_A-12.zip

config文件

  • model_name:模型名称
  • epochs:迭代epoch的数量
  • checkpoint_every:间隔多少步保存一次模型
  • eval_every:间隔多少步验证一次模型
  • learning_rate:学习速率,推荐2e-5, 5e-5, 1e-4
  • sequence_length:序列长度,单GPU时不要超过128
  • batch_size:单GPU时不要超过32
  • num_classes:文本分类的类别数量,若是二分类设置为1
  • warmup_rate:训练时的预热比例,建议0.05, 0.1
  • output_path:输出文件夹,用来存储label_to_index等文件
  • bert_model_path:预训练模型文件夹路径
  • train_data:训练数据路径
  • eval_data:验证数据路径
  • ckpt_model_path:checkpoint模型文件保存路径

空文件

简介

基于深度学习的BERT模型实现中文文本分类 展开 收起
Python
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
Python
1
https://gitee.com/deali/bert-chinese-classifier.git
git@gitee.com:deali/bert-chinese-classifier.git
deali
bert-chinese-classifier
bert-chinese-classifier
master

搜索帮助