NotShrirang/email-spam-filter
Viewer • Updated • 5.17k • 113 • 11
How to use NotShrirang/albert-spam-filter with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="NotShrirang/albert-spam-filter") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("NotShrirang/albert-spam-filter")
model = AutoModelForSequenceClassification.from_pretrained("NotShrirang/albert-spam-filter")This model is a fine-tuned version of albert-base-v2 on this dataset.
The following hyperparameters were used during training:
Base model
albert/albert-base-v2