Token Classification
Transformers
PyTorch
English
roberta
keyphrase-extraction
Eval Results (legacy)
Instructions to use ml6team/keyphrase-extraction-kbir-inspec with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ml6team/keyphrase-extraction-kbir-inspec with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="ml6team/keyphrase-extraction-kbir-inspec")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("ml6team/keyphrase-extraction-kbir-inspec") model = AutoModelForTokenClassification.from_pretrained("ml6team/keyphrase-extraction-kbir-inspec") - Notebooks
- Google Colab
- Kaggle
Commit ·
75baef7
1
Parent(s): dd649ef
Update README.md
Browse files
README.md
CHANGED
|
@@ -88,9 +88,9 @@ class KeyphraseExtractionPipeline(TokenClassificationPipeline):
|
|
| 88 |
**kwargs
|
| 89 |
)
|
| 90 |
|
| 91 |
-
def postprocess(self,
|
| 92 |
results = super().postprocess(
|
| 93 |
-
|
| 94 |
aggregation_strategy=AggregationStrategy.SIMPLE,
|
| 95 |
)
|
| 96 |
return np.unique([result.get("word").strip() for result in results])
|
|
|
|
| 88 |
**kwargs
|
| 89 |
)
|
| 90 |
|
| 91 |
+
def postprocess(self, all_outputs):
|
| 92 |
results = super().postprocess(
|
| 93 |
+
all_outputs=all_outputs,
|
| 94 |
aggregation_strategy=AggregationStrategy.SIMPLE,
|
| 95 |
)
|
| 96 |
return np.unique([result.get("word").strip() for result in results])
|