ncbi/ncbi_disease
Updated • 3.66k • 52
How to use marcopost-it/TaughtNet-disease-chem-gene with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="marcopost-it/TaughtNet-disease-chem-gene") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("marcopost-it/TaughtNet-disease-chem-gene")
model = AutoModelForTokenClassification.from_pretrained("marcopost-it/TaughtNet-disease-chem-gene")Implementation of the model described in "TaughtNet: Learning Multi-Task Biomedical Named Entity Recognition From Single-Task Teachers", published in IEEE Journal of Biomedical and Health Informatics.
This model has been trained for a fewer number of epoch w.r.t. the one described in the paper. Please contact us if you need the complete model.
Paper: https://ieeexplore.ieee.org/document/10041925
Code: https://github.com/marcopost-it/TaughtNet
Email: marco.postiglione@unina.it