Views
No views yet
AutoModelForTokenClassification y Trainer de Hugging Face. Extrae entidades oncológicas con esquema BIO sobre textos médicos en español.1from transformers import pipeline
2
3pipe = pipeline("token-classification", model="jhusef-lopez/ner-clinico-beto-es", aggregation_strategy="simple")
4texto = "El paciente presenta adenocarcinoma de próstata en estadio III."
5print(pipe(texto))| id | etiqueta |
|---|---|
| 0 | B_CANCER_CONCEPT |
| 1 | B_CHEMOTHERAPY |
| 2 | B_DATE |
| 3 | B_DRUG |
| 4 | B_FAMILY |
| 5 | B_FREQ |
| 6 | B_IMPLICIT_DATE |
| 7 | B_INTERVAL |
| 8 | B_METRIC |
| 9 | B_OCURRENCE_EVENT |
| 10 | B_QUANTITY |
| 11 | B_RADIOTHERAPY |
| 12 | B_SMOKER_STATUS |
| 13 | B_STAGE |
| 14 | B_SURGERY |
| 15 | B_TNM |
| 16 | I_CANCER_CONCEPT |
| 17 | I_DATE |
| 18 | I_DRUG |
| 19 | I_FAMILY |
| 20 | I_FREQ |
| 21 | I_IMPLICIT_DATE |
| 22 | I_INTERVAL |
| 23 | I_METRIC |
| 24 | I_OCURRENCE_EVENT |
| 25 | I_SMOKER_STATUS |
| 26 | I_STAGE |
| 27 | I_SURGERY |
| 28 | I_TNM |
| 29 | O |
| Métrica | Valor |
|---|---|
| F1 (validación) | 0.9463 |
| Precision (validación) | 0.9381 |
| Recall (validación) | 0.9546 |
| F1 (test) | 0.9398 |
| Precision (test) | 0.9220 |
| Recall (test) | 0.9584 |
| Batch size (mejor grid) | 8 |
dccuchile/bert-base-spanish-wwm-casedSebastianSalas/biobert_jsonTrainer, EarlyStoppingCallback)