bert_mulit_label_chinese
pip install -r requirements.txt
快速启动
python3 mulit_label.py
训练部分
训练参数
train_batch_size = 16
eval_batch_size = 16
predict_batch_size = 16
learning_rate = 5e-5
num_train_epochs = 35
输入资料
范例档案
data/
- train.tsv
- test.tsv
输出:
0:事物說明
1:他人描述
2:自我探討
f1:
class00_f1 = 0.5
class00_precision = 0.5
class00_recall = 0.5
class01_f1 = 0.8
class01_precision = 0.66666666
评论0
最新资源