eriktks/conll2003
Updated • 41.3k • 166
How to use ai-forever/bert-base-NER-reptile-5-datasets with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="ai-forever/bert-base-NER-reptile-5-datasets") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("ai-forever/bert-base-NER-reptile-5-datasets")
model = AutoModelForTokenClassification.from_pretrained("ai-forever/bert-base-NER-reptile-5-datasets")YAML Metadata Error:"pipeline_tag" must be a string
Model was trained by SberIDP. The pretraining process and technical details are described in this article.
The model is pretrained for NER task using Reptile and can be finetuned for new entities with only a small amount of samples.