在github官网上,我们可以看到通过控制台运行run_classifier.py并加入相关训练参数可以进行模型的训练,在pycharm中,我们可以提前写好训练参数,直接点运行按钮就可以进行模型的训练,操作步骤如下:
1.右击项目中的run_classifier.py文件,选择Edit'run_classifer.py'
2.在Parameters中写入模型训练的参数,路径使用相对路径,在写入参数时需要把github中‘\’符号去掉
data_dir是模型训练和测试数据所在的文件夹
vocab.txt是词典
bert_config_file是模型的一些配置参数
init_checkpoint是预训练的模型