Skip to content

Spam-Detector-AI is a Python package for detecting and filtering spam messages using Machine Learning models.

License

Notifications You must be signed in to change notification settings

adamspd/spam-detection-project

Repository files navigation

Spam-Detector-AI

Tests Current Release Version pypi Version PyPi downloads License GitHub last commit GitHub issues GitHub pull requests GitHub contributors

Spam-Detector-AI is a Python package for detecting and filtering spam messages using Machine Learning models. The package integrates with Django or any other project that uses python and offers different types of classifiers: Naive Bayes, Random Forest, and Support Vector Machine (SVM). Since version 2.1.0, two new classifiers have been added: Logistic Regression and XGBClassifier.

⚠️ Warning: No significant breaking changes were added to the version 2.x.x in terms of usage. On the other hand, the fine-tuning of the models has been moved to a separate module (tuning) and the tests have been moved to a separate module (tests). ⚠️

Table of Contents

Installation

You can install the spam detection package via pip:

pip install spam-detector-ai

Make sure you have the following dependencies installed:

  • scikit-learn
  • nltk
  • pandas
  • numpy
  • joblib
  • xgboost

Additionally, you'll need to download the NLTK data and to do so, use the python interpreter to run the following commands:

import nltk

nltk.download('wordnet')
nltk.download('stopwords')

Usage

Training the Models

Before using the classifiers, you must train the models. Training data is loaded from a CSV file. You can find the training data in the data directory in the GitHub's page of the project. The CSV file must have 3 columns: label, text and label_num. The text column contains the content of the message to analyze and the label column contains the labels ham or spam and label_num contains the number 0 (not spam) or 1(spam).

The more data you have, the better the models will perform.

To train the models, run the following command:

python3 spam_detector_ai/trainer.py

⚠️ Warning: A module not found error may occur ⚠️

If this happens, use an IDE to run the trainer.pyfile until a fix is implemented.

This will train all the models and save them as .joblib files in the models directory. For now, there is 3 models:

  • naive_bayes_model.joblib
  • random_forest_model.joblib
  • svm_model.joblib
  • logistic_regression_model.joblib
  • xgb_model.joblib

Tests

The test results are shown below:

Model: NAIVE_BAYES

Confusion Matrix:
Predicted: Ham Predicted: Spam
Actual: Ham 1935 (True Negative) 170 (False Positive)
Actual: Spam 221 (False Negative) 633 (True Positive)
  • True Negative (TN): 1935 messages were correctly identified as ham (non-spam).
  • False Positive (FP): 170 ham messages were incorrectly identified as spam.
  • False Negative (FN): 221 spam messages were incorrectly identified as ham.
  • True Positive (TP): 633 messages were correctly identified as spam.
Performance Metrics:
Precision Recall F1-Score Support
Ham 0.90 0.92 0.91 2105
Spam 0.79 0.74 0.76 854
Accuracy 0.87 2959
Macro Avg 0.84 0.83 0.84 2959
Weighted Avg 0.87 0.87 0.87 2959
Accuracy: 0.8678607637715444

Model: RANDOM_FOREST

Confusion Matrix:
Predicted: Ham Predicted: Spam
Actual: Ham 2067 (True Negative) 38 (False Positive)
Actual: Spam 36 (False Negative) 818 (True Positive)
  • True Negative (TN): 2067 messages were correctly identified as ham (non-spam).
  • False Positive (FP): 38 ham messages were incorrectly identified as spam.
  • False Negative (FN): 36 spam messages were incorrectly identified as ham.
  • True Positive (TP): 818 messages were correctly identified as spam.
Performance Metrics:
Precision Recall F1-Score Support
Ham 0.98 0.98 0.98 2105
Spam 0.96 0.96 0.96 854
Accuracy 0.97 2959
Macro Avg 0.97 0.97 0.97 2959
Weighted Avg 0.98 0.97 0.98 2959
Accuracy: 0.9749915511997297

Model: SVM

Confusion Matrix:
Predicted: Ham Predicted: Spam
Actual: Ham 2080 (True Negative) 25 (False Positive)
Actual: Spam 41 (False Negative) 813 (True Positive)
  • True Negative (TN): 2080 messages were correctly identified as ham (non-spam).
  • False Positive (FP): 25 ham messages were incorrectly identified as spam.
  • False Negative (FN): 41 spam messages were incorrectly identified as ham.
  • True Positive (TP): 813 messages were correctly identified as spam.
Performance Metrics:
Precision Recall F1-Score Support
Ham 0.98 0.99 0.98 2105
Spam 0.97 0.95 0.96 854
Accuracy 0.98 2959
Macro Avg 0.98 0.97 0.97 2959
Weighted Avg 0.98 0.98 0.98 2959
Accuracy: 0.9773572152754308

Model: LOGISTIC_REGRESSION

Confusion Matrix:
Predicted: Ham Predicted: Spam
Actual: Ham 2065 (True Negative) 48 (False Positive)
Actual: Spam 46 (False Negative) 989 (True Positive)
  • True Negative (TN): 2065 messages were correctly identified as ham (non-spam).
  • False Positive (FP): 48 ham messages were incorrectly identified as spam.
  • False Negative (FN): 46 spam messages were incorrectly identified as ham.
  • True Positive (TP): 989 messages were correctly identified as spam.
Performance Metrics:
Precision Recall F1-Score Support
Ham 0.98 0.98 0.98 2113
Spam 0.95 0.96 0.95 1035
Accuracy 0.97 3148
Macro Avg 0.97 0.97 0.97 3148
Weighted Avg 0.97 0.97 0.97 3148
Accuracy: 0.9707680491551459

Model: XGB

Confusion Matrix:
Predicted: Ham Predicted: Spam
Actual: Ham 2050 (True Negative) 63 (False Positive)
Actual: Spam 28 (False Negative) 1007 (True Positive)
  • True Negative (TN): 2050 messages were correctly identified as ham (non-spam).
  • False Positive (FP): 63 ham messages were incorrectly identified as spam.
  • False Negative (FN): 28 spam messages were incorrectly identified as ham.
  • True Positive (TP): 1007 messages were correctly identified as spam.
Performance Metrics:
Precision Recall F1-Score Support
Ham 0.99 0.97 0.98 2113
Spam 0.94 0.97 0.96 1035
Accuracy 0.97 3148
Macro Avg 0.96 0.97 0.97 3148
Weighted Avg 0.97 0.97 0.97 3148
Accuracy: 0.9710927573062261

The models that performed the best are the SVM and Logistic Regression, with the SVM model achieving slightly higher accuracy than Logistic Regression. Given that no single model achieved perfect accuracy, I have decided to implement a voting classifier. This classifier will combine the predictions of the five models (Naive Bayes, Random Forest, SVM, Logistic Regression, and XGB) using a majority vote system to make the final prediction. This approach aims to leverage the strengths of each model to improve overall prediction accuracy.

Weighted Voting System

To enhance the decision-making process, I've refined our approach to a weighted voting system. This new system assigns different weights to each model's vote based on their respective accuracies. The weights are proportional to the accuracy of each model relative to the sum of the accuracies of all models. The models with higher accuracy have a greater influence on the final decision.

The models and their respective proportional weights are as follows:

  • Naive Bayes: Weight = 0.1822
  • Random Forest: Weight = 0.2047
  • SVM (Support Vector Machine): Weight = 0.2052
  • Logistic Regression: Weight = 0.2039
  • XGBoost (XGB): Weight = 0.2039

These weights were calculated based on the accuracy of each model as a proportion of the total accuracy of all models. The final decision whether a message is spam or not is determined by the weighted spam score. Each model casts a vote (spam or not spam), and this vote is multiplied by the model's weight. The weighted spam scores from all models are then summed up. If this total weighted spam score exceeds 50% of the total possible weight, the message is classified as spam. Otherwise, it's classified as not spam (ham).

This approach ensures that the more accurate models have a larger say in the final decision, thereby increasing the reliability of spam detection. It combines the strengths of each model, compensating for individual weaknesses and provides a more nuanced classification.

System Output

The system provides a detailed output for each message, showing the vote (spam or ham) from each model, along with its weight. It also displays the total weighted spam score and the final classification decision (Spam or Not Spam). This transparency in the voting process allows for easier understanding and debugging of the model's performance on different messages.

If you have trained the models on new data, you can test them by running the following command:

python tests/test.py

⚠️ Warning: A module not found error may occur ⚠️

If this happens, use an IDE to run the test.pyfile until a fix is implemented.

Making Predictions

To use the spam detector in your Django project:

  1. Import the VotingSpamDetector from the prediction module.
  2. Create an instance of the detector.
  3. Use the is_spam method to check if a message is spam.
from spam_detector_ai.prediction.predict import VotingSpamDetector

# Create the spam detector
spam_detector = VotingSpamDetector()

# Check if a message is spam
message = "Enter the message here"
is_spam = spam_detector.is_spam(message)
print(f"Is spam: {is_spam}")

Project Structure

  • classifiers/: Contains the different classifiers (Naive Bayes, Random Forest, SVM, XGB & Logistic Regression).
  • data/: Contains the sample dataset for training the classifiers.
  • loading_and_processing/: Contains utility functions for loading and preprocessing data.
  • models/: Contains the trained models and their vectorizers.
  • prediction/: Contains the main spam detector class.
  • tests/: Contains scripts for testing
  • tuning/: Contains scripts for tuning the classifiers.
  • training/: Contains scripts for training the classifiers.

Contributing

Contributions are welcome! Please read the contributing guidelines before getting started.

License

This project is licensed under the MIT License.

Notes

⚠️ Warning: In a future version, the data/spam.csv file may be removed to lighten the package. ⚠️

The project contains 5 pre-trained models that can be used directly if you want to skip the training step. If you don't want to use the package, you can use the API that I have deployed here.

API Versions

⚠️ Warning: V1 is deprecated and V2 is highly recommended instead. ⚠️

There are two versions of the API:

  • v1: The original version of the API that requires only the message to determine if it is a spam.
  • v2: The updated version of the API that requires a JSON object with additional context about the message.

Why Create a Second Version of the API?

The initial version of the API was too simplistic and did not provide enough context about the message to make an accurate prediction. The new version addresses this limitation by requiring more comprehensive information to improve prediction accuracy.

Required Information for Version 2

To use version 2 of the API, the following information must be provided:

  • text: The content of the message to analyze.
  • subject: The subject of the message (optional).
  • email: The sender's email address.
  • name: The sender's name (optional).
  • user_agent: The sender's user agent string (important but, optional).
  • ip: The sender's IP address.

Additional Checks and Enhancements

In version 2, several additional checks will be performed to enhance the accuracy of spam detection:

  • Email Validation:
    • The email address format is validated.
    • The existence of the domain is verified.
    • MX, DMARC, and SPF records are checked.
    • The email is checked against a list of disposable email addresses.

If the models detect that the message is spam and the email is fake or disposable, the likelihood of the message being spam increases. Similarly, if the IP address is known to be associated with spam activities, the message is more likely to be classified as spam.

  • IP Address Check:

    • The IP address is checked against known spam IP addresses.
  • User Agent Validation:

    • The user agent string is validated.
    • If the user agent appears to be from a bot, the message is more likely to be spam.

By incorporating these additional checks, version 2 of the API provides a more robust and accurate spam detection service.

Example Usage of the API

Version 2

Python Example
import requests


def get_client_ip(request):
    ip = request.META.get('HTTP_X_FORWARDED_FOR')
    if ip:
        ip = ip.split(',')[-1].strip()
    else:
        ip = request.META.get('HTTP_X_REAL_IP') or request.META.get('REMOTE_ADDR')
    return ip


def check_website_contact_form_v2(request):
    message = request.POST.get('message')
    subject = request.POST.get('subject')
    email = request.POST.get('email')
    name = request.POST.get('name', '')  # Optional
    user_agent = request.META.get('HTTP_USER_AGENT', '')  # Optional
    ip = get_client_ip(request)  # Get the IP address of the sender

    # Prepare the JSON payload with the required information
    payload = {
        'text': message,
        'subject': subject,
        'email': email,
        'name': name,
        'user_agent': user_agent,
        'ip': ip
    }

    # Call the spam detection API
    response = requests.post(
            "https://spam-detection-api.adamspierredavid.com/v2/check-spam/",
            json=payload  # Use json parameter to send the payload
    )

    is_spam = False

    # Check if the API request was successful
    if response.status_code == 200:
        # Parse the JSON response
        json_response = response.json()
        is_spam = json_response.get('is_spam')

    if is_spam:
        # Do something if the message is classified as spam.
        # Personally, I save them as a Quarantined object to look at later (knowing that my ML models are not perfect), 
        # and I do nothing with them in my code.
        pass
    else:
        # Do something if the message is not classified as spam
        # Send email, save it to a database as whatever, etc.
        pass
Javascript Example
async function getClientIP(req) {
    let ip = req.headers['x-forwarded-for'];
    if (ip) {
        ip = ip.split(',').pop().trim();
    } else {
        ip = req.headers['x-real-ip'] || req.connection.remoteAddress;
    }
    return ip;
}

async function checkWebsiteContactFormV2(req, res) {
    const message = req.body.message;
    const subject = req.body.subject;
    const email = req.body.email;
    const name = req.body.name || '';  // Optional
    const user_agent = req.headers['user-agent'] || '';  // Optional
    const ip = await getClientIP(req);  // Get the IP address of the sender

    // Prepare the JSON payload with the required information
    const payload = {
        text: message,
        subject: subject,
        email: email,
        name: name,
        user_agent: user_agent,
        ip: ip
    };

    try {
        // Call the spam detection API
        const response = await fetch("https://spam-detection-api.adamspierredavid.com/v2/check-spam/", {
            method: 'POST',
            headers: {
                'Content-Type': 'application/json'
            },
            body: JSON.stringify(payload)
        });

        let is_spam = false;

        // Check if the API request was successful
        if (response.ok) {
            const jsonResponse = await response.json();
            is_spam = jsonResponse.is_spam;
        }

        if (is_spam) {
            // Do something if the message is classified as spam.
            // Personally, I save them as a Quarantined object to look at later (knowing that my ML models are not perfect),
            // and I do nothing with them in my code.
        } else {
            // Do something if the message is not classified as spam
            // Send email, save it to a database as whatever, etc.
        }
    } catch (error) {
        console.error('Error checking spam:', error);
        // Handle error
    }
}

Version 1 DEPRECATED

Python Example
import requests


def check_website_contact_form(request):
    message_ = request.POST.get('message')
    subject = request.POST.get('subject')
    # etc...

    # Concatenate subject and message
    message_with_subject = f'subject: {subject}. {message_}'

    # Call the spam detection API
    response = requests.post(
            "https://spam-detection-api.adamspierredavid.com/v1/check-spam/",
            json={'text': message_with_subject}  # Use json parameter instead of data
            # You can also do `json={'message': message_with_subject}
    )

    is_spam = False

    # Check if the API request was successful
    if response.status_code == 200:
        # Parse the JSON response
        json_response = response.json()
        is_spam = json_response.get('is_spam')

    if is_spam:
        # Do something
        pass
    else:
        # Do something else
        pass
Javascript Example
async function checkWebsiteContactForm(req, res) {
    const message = req.body.message;
    const subject = req.body.subject;

    // Concatenate subject and message
    const messageWithSubject = `subject: ${subject}. ${message}`;

    // Prepare the JSON payload
    const payload = {
        text: messageWithSubject
    };

    try {
        // Call the spam detection API
        const response = await fetch("https://spam-detection-api.adamspierredavid.com/v1/check-spam/", {
            method: 'POST',
            headers: {
                'Content-Type': 'application/json'
            },
            body: JSON.stringify(payload)
        });

        let is_spam = false;

        // Check if the API request was successful
        if (response.ok) {
            const jsonResponse = await response.json();
            is_spam = jsonResponse.is_spam;
        }

        if (is_spam) {
            // Do something
        } else {
            // Do something else
        }
    } catch (error) {
        console.error('Error checking spam:', error);
        // Handle error
    }
}