Skip to content

Kaggle新赛(baseline)-基于BERT的fine-tuning方案+基于tensor2tensor的Transformer Encoder方案

Notifications You must be signed in to change notification settings

zhpmatrix/Kaggle-Quora-Insincere-Questions-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

Kaggle-Quora-Insincere-Questions-Classification

Kaggle新赛-基于BERT的fine-tuning方案baseline

submit.csv是测试集的预测结果

代码和项目基于BERT的中文tagging一样,仅提供关键fine-tuning代码和运行脚本

基于bert的验证集的结果:

class precision recall f1-score
0 0.98 0.98 0.98
1 0.65 0.62 0.63
micro avg 0.96 0.96 0.96
macro avg 0.81 0.80 0.81
weighted avg 0.96 0.96 0.96

基于tensor2tensor的验证集结果:

class precision recall f1-score
0 0.98 0.96 0.96
1 0.23 0.19 0.21
micro avg 0.92 0.92 0.92
macro avg 0.59 0.57 0.58
weighted avg 0.91 0.92 0.91

About

Kaggle新赛(baseline)-基于BERT的fine-tuning方案+基于tensor2tensor的Transformer Encoder方案

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published