SetFit/hate_speech_offensive
Viewer • Updated • 24.8k • 410 • 2
How to use Islamamro/hate-speech-aurora-islamamro with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Islamamro/hate-speech-aurora-islamamro") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Islamamro/hate-speech-aurora-islamamro")
model = AutoModelForSequenceClassification.from_pretrained("Islamamro/hate-speech-aurora-islamamro", device_map="auto")Text classifier fine-tuned from bert-base-uncased on SetFit/hate_speech_offensive (3 classes).
Built, trained and published end-to-end via the Aurora Research Portal by islamamro. Trained on CPU on a 1200-example demo subset.
bert-base-uncased (~109M parameters)This is a demonstration of the Aurora build -> train -> publish pipeline on a small subset, not a production model. Model size here is set by the base architecture, which is varied deliberately across the Aurora showcase to span a range of parameter counts. For production, fine-tune on the full dataset.
Base model
google-bert/bert-base-uncased