metadata
license: apache-2.0
datasets:
- cybersectony/PhishingEmailDetection
library_name: transformers
language:
- en
base_model:
- distilbert/distilbert-base-uncased
tags:
- Phishing
- Email
- URL
- Detection
A distilBERT based Phishing Email Detection Model
Model Overview
This model is specifically fine-tuned for detecting phishing emails using the Hugging Face Trainer API.
Key Specifications
- Base Architecture: DistilBERT
- Task: Multilabel Classification
- Fine-tuning Framework: Hugging Face Trainer API
- Training Duration: 3 epochs
Performance Metrics
- F1-score: 97.717
- Accuracy: 97.716
- Precision: 97.736
- Recall: 97.717
Dataset Details
This model was trained using a Phishing Email Detection Dataset.
Usage Guide Installation
pip install transformers
pip install torch