This repository contains a MATLAB project for EEG hand movement detection using an Artificial Neural Network (ANN) model. The project supports both supervised and semi-supervised learning approaches.
This project aims to detect hand movements based on EEG signals using an ANN model. The dataset used in this project is stored in the Dataset directory. The code consists of several MATLAB files that perform various tasks such as creating the ANN model, training the model using supervised and semi-supervised learning, and testing the model on a separate dataset.
Dataset: Contains the EEG dataset used for training and testing.CreateNet.m: MATLAB script for creating the ANN model.GetMSEs.m: MATLAB script for calculating Mean Squared Errors (MSEs) during training.Main.m: Main MATLAB script that orchestrates the training and testing process.NN.m: MATLAB script defining the ANN architecture and training parameters.SemiSupervised.m: MATLAB script implementing the semi-supervised learning algorithm.Supervised.m: MATLAB script implementing the supervised learning algorithm.
To use this project, follow these steps:
- Clone the repository:
git clone https://github.com/Ilia-Abolhasani/eeg-detection.git - Open MATLAB and navigate to the project directory.
- Run the
Main.mscript to execute the main process. - The script will load the dataset, preprocess the data, create the ANN model, train it using supervised and semi-supervised learning, and test the model on a separate dataset.
- The accuracy of the trained models will be displayed in the MATLAB console.
This project has the following dependencie:
- MATLAB
If you use this code or dataset in your research or project, please consider citing the following paper:
- Dr. Cichocki's Lab (Lab. for Advanced Brain Signal Processing), BSI, RIKEN collaboration with Shanghai Jiao Tong University.
This project is licensed under the MIT License.
For any inquiries or issues regarding this project, please contact Ilia-Abolhasani.