lingyu123-su / multi_label_classification Public

forked from murray-z/multi_label_classification

Notifications You must be signed in to change notification settings
Fork 0
Star 0

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

0 stars 17 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
data		data
bert_multilabel_cls.py		bert_multilabel_cls.py
data_helper.py		data_helper.py
data_preprocess.py		data_preprocess.py
predict.py		predict.py
readme.md		readme.md
requirements.txt		requirements.txt
train.py		train.py

Repository files navigation

基于bert多标签分类

步骤

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

About

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

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%