TestingCapstone/phishing-email-detector-capstone

🧠 Phishing Detection Model (BERT-Large-Uncased)

A transformer-based model fine-tuned to detect phishing content across multiple formats — including emails, URLs, SMS messages, and scripts. Built on BERT-Large-Uncased, it leverages deep contextual understanding of language to classify text as phishing or benign with high accuracy.

📌 Model Details

Base model: bert-large-uncased Architecture: 24 layers • 1024 hidden size • 16 attention heads • ~336M parameters License: Apache 2.0 Language: English Pipeline tag: text-classification

🧩 Model Description

This model was trained to identify phishing-related content by analyzing linguistic and structural patterns commonly found in malicious communications. By leveraging BERT’s bidirectional transformer architecture, it effectively detects phishing attempts even when the message appears legitimate or well-written.

Key Features

• —Detects phishing attempts in text, emails, URLs, and scripts
• —Useful for cybersecurity applications, such as email gateways or web filtering systems
• —Capable of identifying varied phishing tactics (impersonation, link manipulation, credential harvesting, etc.)

🎯 Intended Uses

Recommended use cases:

• —Classify messages, emails, and URLs as phishing or benign
• —Integrate into automated security pipelines, email filtering tools, or chat moderation systems
• —Aid in phishing research or awareness programs

Limitations:

• —May trigger false positives on legitimate content with financial or urgent language
• —Optimized for English text only
• —Should be part of a multi-layered defense strategy, not a standalone cybersecurity control

📊 Evaluation Results

⚙️ Training Details

Hyperparameters

Training Results

🧠 Example Inference

Try the model in Python using the transformers library:

from transformers import pipeline
# Load the phishing detection model
classifier = pipeline("text-classification", model="your-username/phishing-email-detector-capstone")
# Example texts
examples = [
    "Dear colleague, your email storage is full. Click here to verify your account: https://secure-update-login.com",
    "Hi team, the meeting starts at 2 PM today.",
    "You have won a free gift card! Claim now at http://bit.ly/3xYzabc"
]
# Run inference
for text in examples:
    result = classifier(text)[0]
    print(f"Text: {text}\nPrediction: {result['label']} (score: {result['score']:.4f})\n")