Skip to content

RichardG-PU/mlers-ensemble

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

mlers-ensemble

PharmaHacks 2026 — EEG-based AD vs CN classification using a 3-model ensemble.

Models

Model Features LOOCV Description
XGBoost Lasmar 418 (Welch PSD @ 128Hz) ~81% Band power, ratios, Hjorth, spectral entropy
DICE-net RBP + PLV tensors TBD Dual-branch CNN + Transformer, multi-seed
XGBoost v3 1330 (+ entropy + coherence) TBD Nonlinear features + Optuna tuning

Quick Start

pip install -r requirements.txt
python run_all.py

This runs all 3 models (LOOCV + final training + test predictions) and combines them via weighted ensemble. Final output: predictions.csv.

Running Individual Models

# XGBoost Lasmar
python models/xgboost_lasmar/train.py

# DICE-net (requires precomputed features)
python models/dicenet/precompute_features.py
python models/dicenet/predict.py

# XGBoost v3
python models/xgboost_v3/train.py

# Ensemble (after all models have run)
python ensemble.py

Data

  • training/AD/ — 25 AD subjects (.npy, 19 channels)
  • training/CN/ — 28 CN subjects (13 labeled)
  • training/test/ — 14 unlabeled test subjects
  • training/train_label_mapping.csv — subject labels

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages