Downloads · 30 days
4
3% of all-time downloads
NicoCasso/finetuned-medical-fr
finetuned-medical-fr is a text classification model from NicoCasso. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
Modèle de classification fine-tuné sur des textes médicaux en français. Il a été entraîné avec 🤗 Transformers et PyTorch à partir d’un modèle pré-entraîné.
Downloads · 30 days
4
3% of all-time downloads
All-time downloads
137
Public
Parameters
111M
1.3 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors443 MB · 99%
From the Hugging Face model README
Modèle de classification fine-tuné sur des textes médicaux en français. Il a été entraîné avec 🤗 Transformers et PyTorch à partir d’un modèle pré-entraîné.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("NicoCasso/finetuned-medical-fr")
model = AutoModelForSequenceClassification.from_pretrained("NicoCasso/finetuned-medical-fr")
text = "Le patient présente une douleur abdominale persistante."
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
outputs = model(**inputs)
prediction = outputs.logits.argmax(dim=-1).item()
print("Classe prédite :", prediction)
🔬 Détails Langue : Français 🇫🇷 Tâche : Classification de texte médical Framework : PyTorch Base de données : Jeu de données personnalisé (non public)
📄 Licence Ce modèle est publié sous la licence Apache 2.0 : ✅ libre d’usage, modification, et distribution, même à des fins commerciales.
Pour plus d’infos : https://www.apache.org/licenses/LICENSE-2.0
✨ Auteur Ce modèle a été entraîné et publié par NicoCasso.