ucirvine/sms_spam
Viewer โข Updated โข 5.57k โข 5.57k โข 55
How to use wesleyacheng/sms-spam-classification-with-bert with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="wesleyacheng/sms-spam-classification-with-bert") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("wesleyacheng/sms-spam-classification-with-bert")
model = AutoModelForSequenceClassification.from_pretrained("wesleyacheng/sms-spam-classification-with-bert")First posted in my Kaggle.
You know what really grinds my gears. Spam! ๐ค
I made a sms spam classifier using transfer learning on BERT with a Singaporean SMS Spam dataset.