Downloads · 30 days
20
41% of all-time downloads
tmt3103/BioASQ-yesno-PudMedBERT
BioASQ-yesno-PudMedBERT is a text classification model from tmt3103. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
--- - Model architecture: BERT - Pretrained base: microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext - Fine-tuned on: BioASQ Phase B Yes/No question dataset - Task type: Binary classification (Yes/No) - Inp…
Downloads · 30 days
20
41% of all-time downloads
All-time downloads
49
Public
Parameters
109M
438 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors438 MB · 100%
From the Hugging Face model README
microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext| Metric | Value |
|---|---|
| Accuracy | 91.44% |
| F1 Score | 89.36% |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
# Load tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("tmt3103/BioASQ-yesno-PudMedBERT")
model = AutoModelForSequenceClassification.from_pretrained("tmt3103/BioASQ-yesno-PudMedBERT")
def predict_yesno(question: str, context: str) -> str:
inputs = tokenizer(question, context, truncation=True, padding=True, return_tensors="pt")
outputs = model(**inputs)
probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
return "Yes" if probs[0][1] > probs[0][0] else "No"
# Example usage
question = "Does aspirin reduce inflammation?"
context = "Aspirin is widely used as an anti-inflammatory medication in clinical practice."
print(f"Question: {question}\nPredicted answer: {predict_yesno(question, context)}")
For questions, issues, or collaboration inquiries, please contact: