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-xlmr-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.9499 |
| Precision (validación) | 0.9394 |
| Recall (validación) | 0.9607 |
| F1 (test) | 0.9426 |
| Precision (test) | 0.9261 |
| Recall (test) | 0.9597 |
| Batch size (mejor grid) | 8 |
FacebookAI/xlm-roberta-largeSebastianSalas/biobert_jsonTrainer, EarlyStoppingCallback)