/multi_label_classification

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

Primary LanguagePython

基于bert多标签分类

步骤

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