badmatr11x/hate-offensive-speech
Viewer • Updated • 56.7k • 60 • 5
How to use badmatr11x/distilroberta-base-offensive-hateful-speech-text-multiclassification with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="badmatr11x/distilroberta-base-offensive-hateful-speech-text-multiclassification") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("badmatr11x/distilroberta-base-offensive-hateful-speech-text-multiclassification")
model = AutoModelForSequenceClassification.from_pretrained("badmatr11x/distilroberta-base-offensive-hateful-speech-text-multiclassification", device_map="auto")This is pre-trained model for offensive and hateful speech multiclassification. Model is fine-tuned on distilroberta-base model from the huggingface repository. This model is pre-trained on my original datasets. Reported accuracy of this model is 94.50%.
You can find all the datasets on this repository. Find out my Space on this model here.
You can find training notebook on my github profile @purveshpatel511. Report any bugs or issue here.
Checkout my GitHub Profile @purveshpatel511.