stacking-ensemble-learning

IDS Stacking Ensemble Learning

Stacking ensemble (XGBoost + CatBoost + LightGBM + AdaBoost) for base mdoel with Random Forest as meta-model for Network Intrusion Detection System (IDS).

Metrics (test set)

Requirements

pip install scikit-learn xgboost catboost lightgbm pandas numpy joblib huggingface_hub

Usage

import joblib
import numpy as np
from huggingface_hub import hf_hub_download

models = {k: joblib.load(hf_hub_download("mrsindhunugroho/stacking-ensemble-learning", f"models/{k}_model.pkl"))
          for k in ["xgboost", "catboost", "lightgbm", "adaboost"]}
meta = joblib.load(hf_hub_download("mrsindhunugroho/stacking-ensemble-learning", "models/meta_model.pkl"))
le   = joblib.load(hf_hub_download("mrsindhunugroho/stacking-ensemble-learning", "models/label_encoder.pkl"))

base_preds = np.column_stack([m.predict(X) for m in models.values()])
y_pred = le.inverse_transform(meta.predict(base_preds))

Dataset

• Source: Kaggle — CSE-CIC-IDS2018 Cleaned
• Original: Canadian Institute for Cybersecurity (CSE-CIC-IDS2018)
• Preprocessing: sampling, label encoding, imputation, feature sanitization

Author

Sindhu Nugroho — ORCID