model update
Browse files
README.md
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
---
|
| 2 |
datasets:
|
| 3 |
-
- btc
|
| 4 |
metrics:
|
| 5 |
- f1
|
| 6 |
- precision
|
|
@@ -12,9 +12,9 @@ model-index:
|
|
| 12 |
name: Token Classification
|
| 13 |
type: token-classification
|
| 14 |
dataset:
|
| 15 |
-
name: btc
|
| 16 |
-
type: btc
|
| 17 |
-
args: btc
|
| 18 |
metrics:
|
| 19 |
- name: F1
|
| 20 |
type: f1
|
|
@@ -78,6 +78,18 @@ For F1 scores, the confidence interval is obtained by bootstrap as below:
|
|
| 78 |
Full evaluation can be found at [metric file of NER](https://huggingface.co/tner/roberta-large-btc/raw/main/eval/metric.json)
|
| 79 |
and [metric file of entity span](https://huggingface.co/tner/roberta-large-btc/raw/main/eval/metric_span.json).
|
| 80 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 81 |
|
| 82 |
### Training hyperparameters
|
| 83 |
|
|
|
|
| 1 |
---
|
| 2 |
datasets:
|
| 3 |
+
- tner/btc
|
| 4 |
metrics:
|
| 5 |
- f1
|
| 6 |
- precision
|
|
|
|
| 12 |
name: Token Classification
|
| 13 |
type: token-classification
|
| 14 |
dataset:
|
| 15 |
+
name: tner/btc
|
| 16 |
+
type: tner/btc
|
| 17 |
+
args: tner/btc
|
| 18 |
metrics:
|
| 19 |
- name: F1
|
| 20 |
type: f1
|
|
|
|
| 78 |
Full evaluation can be found at [metric file of NER](https://huggingface.co/tner/roberta-large-btc/raw/main/eval/metric.json)
|
| 79 |
and [metric file of entity span](https://huggingface.co/tner/roberta-large-btc/raw/main/eval/metric_span.json).
|
| 80 |
|
| 81 |
+
### Usage
|
| 82 |
+
This model can be used through the [tner library](https://github.com/asahi417/tner). Install the library via pip
|
| 83 |
+
```shell
|
| 84 |
+
pip install tner
|
| 85 |
+
```
|
| 86 |
+
and activate model as below.
|
| 87 |
+
```python
|
| 88 |
+
from tner import TransformersNER
|
| 89 |
+
model = TransformersNER("tner/roberta-large-btc")
|
| 90 |
+
model.predict(["Jacob Collier is a Grammy awarded English artist from London"])
|
| 91 |
+
```
|
| 92 |
+
It can be used via transformers library but it is not recommended as CRF layer is not supported at the moment.
|
| 93 |
|
| 94 |
### Training hyperparameters
|
| 95 |
|