ealvaradob/phishing-dataset
Updated • 891 • 63
How to use ealvaradob/bert-phishing-text with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="ealvaradob/bert-phishing-text") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ealvaradob/bert-phishing-text")
model = AutoModelForSequenceClassification.from_pretrained("ealvaradob/bert-phishing-text")WARNING ...
This is NOT the final BERT model trained for phishing detection. It only corresponds to an evaluation of BERT performance against email and SMS samples.
This model has the following performance in email and SMS phishing detection:
👇¡CHECK BERT FINAL MODEL FINETUNED FOR PHISHING DETECTION ON THE FOLLOWING LINK!👇