文本
texta-bert-tagger的Python项目详细描述
TEXTA Bert Tagger Python包
安装
来自Git
pip install git+https://git.texta.ee/texta/texta-bert-tagger-python.git
测试
python -m pytest -v tests
文档
文档可用here。在
使用
微调BERT模型
>>> from texta_bert_tagger.tagger import BertTagger
>>> bert_tagger = BertTagger()
>>> data_sample = {"good": ["Täna on ilus ilm.", "Kuidas käsi käib?"], "bad": ["Arno on loll.", "ei"]}
# Train model
>>> bert_tagger.train(data_sample, n_epochs=1)
# Predict
>>> bert_tagger.tag_text("Ei hooli.")
'bad'
- 项目
标签: