bert 分類模型

大神的帖子：https://blog.csdn.net/xmxoxo/article/details/89315370

github: https://github.com/xmxoxo/BERT-train2deploy

記錄本人訓練情況：

1.數據：train.tsv,test.tsv ,dev.tsv

2.改源碼：-> 修改run_classifier.py文件中get_labels等方法

3.保存label2id的字典

4.使用命令訓練模型

5.預測

6.後面使用的模型服務端，使用的是.pb格式的模型文件，所以需要把生成的ckpt格式模型文件轉換成.pb格式的模型文件。

7. 部署服務器bert-base-serving-start -model_dir /icooper/output/mobile_0 -bert_model_dir /icooper/chinese_L-12_H-768_A-12 -model_pb_dir /icooper/output/mobile_0 -mode CLASS -max_seq_len 200 -port 5575 -port_out 5576 -http_port 8091

新手踩點:

1.當時在docker中部署環境LANG="" 編碼不對，locale查看，改LANG=en_US.UTF-8

2.沒保存pd文件bert-base-serving-start運行出來有莫名其妙的錯誤，保存後正確

果然跟大神走是沒錯的！！！