multi_label_classification icon indicating copy to clipboard operation
multi_label_classification copied to clipboard

Published 20 hours ago • verified publisher icon murray-z
→

Metadata

基于pytorch + bert的多标签文本分类（multi label text classification）

基于bert多标签分类

步骤

  • 数据样例见data/train.json
  • 获得label2idx.json python data_preprocess.py
  • 训练 python train.py
  • 预测 pyhton predict.py

← Metadata

73
Stars
15
Forks
Watchers

Owner

verified publisher icon murray-z

Metadata

基于pytorch + bert的多标签文本分类（multi label text classification）

Back