Instructions to use KBLab/bert-base-swedish-lowermix-reallysimple-ner with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use KBLab/bert-base-swedish-lowermix-reallysimple-ner with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="KBLab/bert-base-swedish-lowermix-reallysimple-ner")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("KBLab/bert-base-swedish-lowermix-reallysimple-ner") model = AutoModelForTokenClassification.from_pretrained("KBLab/bert-base-swedish-lowermix-reallysimple-ner", device_map="auto") - Notebooks
- Google Colab
- Kaggle
metadata
model:
- KB/bert-base-swedish-cased
tags:
- token-classification
- sequence-tagger-model
- bert
language: sv
datasets:
- KBLab/sucx3_ner
widget:
- text: Emil bor i Lönneberga
KB-BERT for NER
Mixed cased and uncased data
This model is based on KB-BERT and was fine-tuned on the SUCX 3.0 - NER corpus, using the simple tags and partially lowercased data. For this model we used a variation of the data that did not use BIO-encoding to differentiate between the beginnings (B), and insides (I) of named entity tags.
The model was trained on the training data only, with the best model chosen by its performance on the validation data. You find more information about the model and the performance on our blog: https://kb-labb.github.io/posts/2022-02-07-suc31