两个模型随便使用。
比赛链接:
使用bert提取特征,Bidirectional 训练模型,可以达到 74.39565。需要一定的内存。
pip install keras
pip install keras_bert
# 指定数据目录,
data_dir = './data/'
# 将bert预训练数据放到改目录
# https://storage.googleapis.com/bert_models/2018_11_03/chinese_L-12_H-768_A-12.zip
BERT_PRETRAINED_DIR = './chinese_L-12_H-768_A-12/'
# 运行
python3 xf_bert_v2.py
- 下载好数据存放
/data文件夹下,创建data/xfyun文件夹 - 数据预处理生成数据集
python xunfei/pre_classify.py - 训练模型
python xunfei_classify.py -d ./data/xfyun -n xunfei -m train - 预测模型
python xunfei_classify.py -d ./data/xfyun -n xunfei -m predict
@Galen20190717