Token Classification
Transformers
PyTorch
English
roberta
keyphrase-extraction
Eval Results (legacy)
Instructions to use ml6team/keyphrase-extraction-kbir-kptimes with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ml6team/keyphrase-extraction-kbir-kptimes with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="ml6team/keyphrase-extraction-kbir-kptimes")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("ml6team/keyphrase-extraction-kbir-kptimes") model = AutoModelForTokenClassification.from_pretrained("ml6team/keyphrase-extraction-kbir-kptimes", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
baceaac
1
Parent(s): 306e18b
Update README.md
Browse files
README.md
CHANGED
|
@@ -47,7 +47,7 @@ Here is where Artificial Intelligence 🤖 comes in. Currently, classical machin
|
|
| 47 |
|
| 48 |
|
| 49 |
## 📓 Model Description
|
| 50 |
-
This model uses [KBIR](https://huggingface.co/
|
| 51 |
You can find more information about the architecture in this [paper](https://arxiv.org/abs/2112.08547).
|
| 52 |
|
| 53 |
Keyphrase extraction models are transformer models fine-tuned as a token classification problem where each word in the document is classified as being part of a keyphrase or not.
|
|
|
|
| 47 |
|
| 48 |
|
| 49 |
## 📓 Model Description
|
| 50 |
+
This model uses [KBIR](https://huggingface.co/bloomberg/KBIR) as its base model and fine-tunes it on the [KPTimes dataset](https://huggingface.co/datasets/midas/kptimes). KBIR or Keyphrase Boundary Infilling with Replacement is a pre-trained model which utilizes a multi-task learning setup for optimizing a combined loss of Masked Language Modeling (MLM), Keyphrase Boundary Infilling (KBI) and Keyphrase Replacement Classification (KRC).
|
| 51 |
You can find more information about the architecture in this [paper](https://arxiv.org/abs/2112.08547).
|
| 52 |
|
| 53 |
Keyphrase extraction models are transformer models fine-tuned as a token classification problem where each word in the document is classified as being part of a keyphrase or not.
|