CoolFace
Modelpublic

sobanahmed6061/ddos-mitigation-models

sourceHugging Faceupdated 5mo agoView on Hugging Face
0likes1downloads
Model Card

DDoS Mitigation Tool — ML Models

Pre-trained models for DDoS attack detection and classification.

Models Included

FileModelPurposePerformance
isolation_forest.joblibIsolation ForestAnomaly detectionF1: 0.9485
lstm_model.kerasLSTM Neural NetworkTemporal analysisF1: 0.9804
classifier_rf.joblibRandom ForestAttack classificationAcc: 100%
scaler.joblibStandardScalerIF feature scaling
lstm_scaler.joblibMinMaxScalerLSTM feature scaling
classifier_scaler.joblibStandardScalerRF feature scaling
classifier_encoder.joblibLabelEncoderClass encoding
model_meta.jsonMetadataIF model info
lstm_meta.jsonMetadataLSTM model info
classifier_meta.jsonMetadataRF model info

Attack Types Detected

  • SYN Flood
  • UDP Flood
  • ICMP Flood
  • HTTP Flood
  • Slowloris
  • Volumetric

Training Data

  • Normal traffic: 169 windows
  • Attack traffic: 133 windows (5 attack types)
  • Window size: 5 seconds
  • Features: 20 per window

Usage

python
import joblib
import numpy as np

# Load models
iso_model  = joblib.load('isolation_forest.joblib')
iso_scaler = joblib.load('scaler.joblib')

# Predict (1=normal, -1=anomaly)
features = np.array([[...]])  # 20 features
x_scaled = iso_scaler.transform(features)
prediction = iso_model.predict(x_scaled)

Full Project

GitHub: https://github.com/sobanahmed6061/ddos-mitigation-tool