diogocarapito/text-to-icpc2
Viewer • Updated • 18.1k • 14
How to use diogocarapito/text-to-icpc2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="diogocarapito/text-to-icpc2") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("diogocarapito/text-to-icpc2")
model = AutoModelForSequenceClassification.from_pretrained("diogocarapito/text-to-icpc2")This model aims to classify a diagnosis into an ICPC-2 code.
The model was trained on health diagnosis data and ICPC-2 codes, as well as mappings to ICD-10 and data augmentation usingo Open AI's gpt-4.
This project was developed by Diogo Carapito with the support of a research grant from AICIB and APMGF, as part of the Family Medicine residency program.
Base model
google-bert/bert-base-uncased