qurator-spk/sbb_ner
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_ner.git synced 2026-02-26 10:21:56 +01:00
Code Issues Releases Wiki Activity

move bert pre-training code to sbb_ner

Browse source
This commit is contained in:
Kai 2022-02-21 15:41:27 +01:00
parent 9cc644edac
commit c2ff011010
5 changed files with 834 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
setup.py
View file

@ -25,7 +25,11 @@ setup(
"compile_conll=qurator.sbb_ner.ground_truth.conll:main",
"compile_wikiner=qurator.sbb_ner.ground_truth.wikiner:main",
"join-gt=qurator.sbb_ner.ground_truth.join_gt:main",
"bert-ner=qurator.sbb_ner.models.bert:main"
"bert-ner=qurator.sbb_ner.models.bert:main",
"collectcorpus=qurator.sbb_ner.models.corpus:collect",
"bert-pregenerate-trainingdata=qurator.sbb_ner.models.pregenerate_training_data:main",
"bert-finetune=qurator.sbb_ner.models.finetune_on_pregenerated:main"
]
},
python_requires='>=3.6.0',