使用BERT模型做文本分类;面向工业用途
使用BERT模型做文本分类;面向工业用途
如果仅对test.txt执行预测,只需要把 do_predict 设为True,do_train 与do_eval 设置为false。
如果上面步骤进展顺利,恭喜,在输出目录内已经生成了训练和评估结果。屏幕上也打印出了评估准确率等。
bert-base-serving-start \ -model_dir ./ \ # 训练输出目录【主要是其他Mode用到】。 -bert_model_dir F:\chinese_L-12_H-768_A-12 # BERT自身的目录。 -model_pb_dir ./ \ # pb模型路径,就填./ 当前目录就行了。。 -mode CLASS \ # 运行模式,分类就是CLASS -max_seq_len 200 \ # 最大序列长度。要跟训练时参数一致。