Token Classification
Transformers
Safetensors
English
roberta
feature-extraction
entity-recognition
foundation-model
RoBERTa
generic
Instructions to use numind/NuNER-v1.0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use numind/NuNER-v1.0 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="numind/NuNER-v1.0")# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("numind/NuNER-v1.0") model = AutoModel.from_pretrained("numind/NuNER-v1.0", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -90,4 +90,15 @@ emb = torch.cat(
|
|
| 90 |
|
| 91 |
# for better speed
|
| 92 |
# emb = output.hidden_states[-1]
|
| 93 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
|
| 91 |
# for better speed
|
| 92 |
# emb = output.hidden_states[-1]
|
| 93 |
+
```
|
| 94 |
+
|
| 95 |
+
## Citation
|
| 96 |
+
|
| 97 |
+
@misc{bogdanov2024nuner,
|
| 98 |
+
title={NuNER: Entity Recognition Encoder Pre-training via LLM-Annotated Data},
|
| 99 |
+
author={Sergei Bogdanov and Alexandre Constantin and Timothée Bernard and Benoit Crabbé and Etienne Bernard},
|
| 100 |
+
year={2024},
|
| 101 |
+
eprint={2402.15343},
|
| 102 |
+
archivePrefix={arXiv},
|
| 103 |
+
primaryClass={cs.CL}
|
| 104 |
+
}
|