# !pip install -U keras tensorflow huggingface_hub
# Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here;
# "jax" and "torch" also work for computation once TensorFlow is installed.
import os
os.environ["KERAS_BACKEND"] = "tensorflow"
import keras
model = keras.saving.load_model("hf://Georgios-Ak/EEG-Classification-Autoencoder-CNN")
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
EEG-Based Biometric Identification Model (Autoencoder + CNN)
This model implements a hybrid architecture combining an Autoencoder for feature extraction and a Convolutional Neural Network (CNN) for classification of EEG signals. It is designed for biometric identification using spectrogram-transformed EEG data.
Model Overview
- Input: Spectrograms generated from EEG signals.
- Architecture:
- Autoencoder: Compresses high-dimensional spectrogram data into compact latent representations.
- CNN Classifier: Learns patterns from either raw spectrograms or encoded features for classification.
- Training Dataset: Public EEG Motor Movement/Imagery Dataset (BCI2000), including signals from 109 subjects across 14 tasks.
Performance
The combined Autoencoder + CNN approach achieves significantly improved classification accuracy compared to baseline CNN-only models, with performance metrics including:
- Accuracy: Up to 99.6%
- F1 Score: High across all subject classes
- Downloads last month
- -
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
# Gated model: Login with a HF token with gated access permission hf auth login