vector-institute/newsmediabias-plus
Viewer • Updated • 40.9k • 25 • 9
How to use maximuspowers/nmbp-bert-headlines with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="maximuspowers/nmbp-bert-headlines") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("maximuspowers/nmbp-bert-headlines")
model = AutoModelForSequenceClassification.from_pretrained("maximuspowers/nmbp-bert-headlines", device_map="auto")Classifies sentences as "Likely" or "Unlikely" biased/disinformation (max token len 128).
Fine-tuned BERT (bert-base-uncased) on the headline and text_label fields in the News Media Bias Plus Dataset.
This model was trained without weighted sampling, and the dataset contains 81.9% 'Likely' and 18.1% 'Unlikely' examples. The same model trained with weighted sampling preformed better when evaluated by gpt-4o-mini as a judge and is available here.
Evaluated on a 0.1 random sample of the NMB+ dataset, unseen during training
from transformers import pipeline
classifier = pipeline("text-classification", model="maximuspowers/nmbp-bert-headlines")
result = classifier("He was a terrible politician.", top_k=2)
[
{
'label': 'Likely',
'score': 0.9967995882034302
},
{
'label': 'Unlikely',
'score': 0.003200419945642352
}
]
Base model
google-bert/bert-base-uncased