bigbio/bc5cdr
Updated • 2.19k • 19
How to use yam89421/app-v3-pubmedbert-bc5cdr with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="yam89421/app-v3-pubmedbert-bc5cdr") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("yam89421/app-v3-pubmedbert-bc5cdr")
model = AutoModelForTokenClassification.from_pretrained("yam89421/app-v3-pubmedbert-bc5cdr", device_map="auto")Modèle de la méthode apprise APP-v3 du projet NLP en santé (ESIEE Paris, E5-BIO, AMARA Yanis / TOMAS Alban) : https://github.com/yam89421/NLP-SANTE-BC5CDR
microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext (PubMedBERT).Chemical et Disease (étiquettes O, B-/I-Chemical, B-/I-Disease).2.4c_methode_apprise_v3.ipynb).Prédiction : textes découpés en mots comme dans le projet (tals.tokeniser_avec_offsets), tokenizer en minuscules,
premier sous-mot de chaque mot, décodage BIO « reparer » (src/apprise.py).