Commit
·
c32e389
1
Parent(s):
a16df5d
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 [OpenKP dataset](https://huggingface.co/datasets/midas/openkp). 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.
|