文件名称:Bert-Multi-Label-Text-Classification:此存储库包含用于多标签文本分类的预训练BERT模型的PyTorch实现
文件大小:154KB
文件格式:ZIP
更新时间:2024-02-24 12:04:37
nlp text-classification transformers pytorch multi-label-classification
PyTorch的Bert多标签文本分类 此仓库包含用于多标签文本分类的预训练BERT和XLNET模型的PyTorch实现。 代码结构 在项目的根目录,您将看到: ├── pybert | └── callback | | └── lrscheduler.py | | └── trainingmonitor.py | | └── ... | └── config | | └── basic_config.py #a configuration file for storing model parameters | └── dataset | └── io
【文件预览】:
Bert-Multi-Label-Text-Classification-master
----Pipfile(253B)
----LICENSE(1KB)
----run_bert.py(12KB)
----pybert()
--------train()
--------model()
--------test()
--------pretrain()
--------preprocessing()
--------output()
--------dataset()
--------__init__.py(15B)
--------io()
--------common()
--------configs()
--------callback()
----__init__.py(15B)
----.idea()
--------misc.xml(174B)
--------vcs.xml(180B)
--------deployment.xml(855B)
--------modules.xml(535B)
--------encodings.xml(138B)
--------.gitignore(39B)
--------Bert-Multi-Label-Text-Classification.iml(561B)
--------inspectionProfiles()
----run_albert.py(12KB)
----requirements.txt(485B)
----.gitignore(1KB)
----README.md(5KB)
----run_xlnet.py(12KB)
----predict_one.py(1KB)