├── .gitignore ├── .gitmodules ├── DL_Models ├── Ensemble.py ├── tf_models │ ├── BaseNet.py │ ├── CNN │ │ ├── CNN.py │ │ └── __init__.py │ ├── ConvNet.py │ ├── EEGNet │ │ ├── EEGModels.py │ │ ├── __init__.py │ │ └── eegNet.py │ ├── Ensemble_tf.py │ ├── InceptionTime │ │ ├── Inception.py │ │ └── __init__.py │ ├── PyramidalCNN │ │ ├── PyramidalCNN.py │ │ └── __init__.py │ ├── Xception │ │ └── Xception.py │ ├── __init__.py │ └── utils │ │ ├── __init__.py │ │ ├── losses.py │ │ └── plot.py └── torch_models │ ├── BaseNetTorch.py │ ├── CNN │ ├── CNN.py │ └── __init__.py │ ├── ConvNetTorch.py │ ├── EEGNet │ ├── __init__.py │ └── eegNet.py │ ├── Ensemble_torch.py │ ├── InceptionTime │ ├── InceptionTime.py │ └── __init__.py │ ├── Modules.py │ ├── PyramidalCNN │ ├── PyramidalCNN.py │ └── __init__.py │ ├── Xception │ ├── Xception.py │ └── __init__.py │ ├── __init__.py │ └── torch_utils │ ├── __init__.py │ ├── custom_losses.py │ ├── dataloader.py │ ├── plot.py │ ├── training.py │ └── utils.py ├── LICENSE ├── README.md ├── StandardML_Models ├── StandardClassifier_1D.py ├── StandardRegressor_1D.py └── StandardRegressor_2D.py ├── benchmark.py ├── config.py ├── eegeyenet_benchmark.yml ├── general_requirements.txt ├── hyperparameters.py ├── main.py ├── results └── standardML │ └── 1622883925 │ ├── config.csv │ ├── console.out │ ├── info.log │ ├── runs.csv │ └── statistics.csv ├── standard_ml_requirements.txt ├── tensorflow_requirements.txt ├── tuning ├── tuningDL │ └── kerasTuner.py └── tuningML │ ├── ml_utils.py │ ├── sklearnclassifier.py │ ├── sklearndummy.py │ └── sklearnregressor.py └── utils ├── IOHelper.py ├── __init__.py ├── tables_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/.gitmodules -------------------------------------------------------------------------------- /DL_Models/Ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/Ensemble.py -------------------------------------------------------------------------------- /DL_Models/tf_models/BaseNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/BaseNet.py -------------------------------------------------------------------------------- /DL_Models/tf_models/CNN/CNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/CNN/CNN.py -------------------------------------------------------------------------------- /DL_Models/tf_models/CNN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/tf_models/ConvNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/ConvNet.py -------------------------------------------------------------------------------- /DL_Models/tf_models/EEGNet/EEGModels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/EEGNet/EEGModels.py -------------------------------------------------------------------------------- /DL_Models/tf_models/EEGNet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/tf_models/EEGNet/eegNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/EEGNet/eegNet.py -------------------------------------------------------------------------------- /DL_Models/tf_models/Ensemble_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/Ensemble_tf.py -------------------------------------------------------------------------------- /DL_Models/tf_models/InceptionTime/Inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/InceptionTime/Inception.py -------------------------------------------------------------------------------- /DL_Models/tf_models/InceptionTime/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/tf_models/PyramidalCNN/PyramidalCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/PyramidalCNN/PyramidalCNN.py -------------------------------------------------------------------------------- /DL_Models/tf_models/PyramidalCNN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/tf_models/Xception/Xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/Xception/Xception.py -------------------------------------------------------------------------------- /DL_Models/tf_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/tf_models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/tf_models/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/utils/losses.py -------------------------------------------------------------------------------- /DL_Models/tf_models/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/tf_models/utils/plot.py -------------------------------------------------------------------------------- /DL_Models/torch_models/BaseNetTorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/BaseNetTorch.py -------------------------------------------------------------------------------- /DL_Models/torch_models/CNN/CNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/CNN/CNN.py -------------------------------------------------------------------------------- /DL_Models/torch_models/CNN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/torch_models/ConvNetTorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/ConvNetTorch.py -------------------------------------------------------------------------------- /DL_Models/torch_models/EEGNet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/torch_models/EEGNet/eegNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/EEGNet/eegNet.py -------------------------------------------------------------------------------- /DL_Models/torch_models/Ensemble_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/Ensemble_torch.py -------------------------------------------------------------------------------- /DL_Models/torch_models/InceptionTime/InceptionTime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/InceptionTime/InceptionTime.py -------------------------------------------------------------------------------- /DL_Models/torch_models/InceptionTime/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/torch_models/Modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/Modules.py -------------------------------------------------------------------------------- /DL_Models/torch_models/PyramidalCNN/PyramidalCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/PyramidalCNN/PyramidalCNN.py -------------------------------------------------------------------------------- /DL_Models/torch_models/PyramidalCNN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/torch_models/Xception/Xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/Xception/Xception.py -------------------------------------------------------------------------------- /DL_Models/torch_models/Xception/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/torch_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/torch_models/torch_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DL_Models/torch_models/torch_utils/custom_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/torch_utils/custom_losses.py -------------------------------------------------------------------------------- /DL_Models/torch_models/torch_utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/torch_utils/dataloader.py -------------------------------------------------------------------------------- /DL_Models/torch_models/torch_utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/torch_utils/plot.py -------------------------------------------------------------------------------- /DL_Models/torch_models/torch_utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/torch_utils/training.py -------------------------------------------------------------------------------- /DL_Models/torch_models/torch_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/DL_Models/torch_models/torch_utils/utils.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/README.md -------------------------------------------------------------------------------- /StandardML_Models/StandardClassifier_1D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/StandardML_Models/StandardClassifier_1D.py -------------------------------------------------------------------------------- /StandardML_Models/StandardRegressor_1D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/StandardML_Models/StandardRegressor_1D.py -------------------------------------------------------------------------------- /StandardML_Models/StandardRegressor_2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/StandardML_Models/StandardRegressor_2D.py -------------------------------------------------------------------------------- /benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/benchmark.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/config.py -------------------------------------------------------------------------------- /eegeyenet_benchmark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/eegeyenet_benchmark.yml -------------------------------------------------------------------------------- /general_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/general_requirements.txt -------------------------------------------------------------------------------- /hyperparameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/hyperparameters.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/main.py -------------------------------------------------------------------------------- /results/standardML/1622883925/config.csv: -------------------------------------------------------------------------------- 1 | LR_task 2 | antisaccade 3 | max 4 | -------------------------------------------------------------------------------- /results/standardML/1622883925/console.out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /results/standardML/1622883925/info.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/results/standardML/1622883925/info.log -------------------------------------------------------------------------------- /results/standardML/1622883925/runs.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/results/standardML/1622883925/runs.csv -------------------------------------------------------------------------------- /results/standardML/1622883925/statistics.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/results/standardML/1622883925/statistics.csv -------------------------------------------------------------------------------- /standard_ml_requirements.txt: -------------------------------------------------------------------------------- 1 | xgboost -------------------------------------------------------------------------------- /tensorflow_requirements.txt: -------------------------------------------------------------------------------- 1 | tensorflow-gpu 2 | keras -------------------------------------------------------------------------------- /tuning/tuningDL/kerasTuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/tuning/tuningDL/kerasTuner.py -------------------------------------------------------------------------------- /tuning/tuningML/ml_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/tuning/tuningML/ml_utils.py -------------------------------------------------------------------------------- /tuning/tuningML/sklearnclassifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/tuning/tuningML/sklearnclassifier.py -------------------------------------------------------------------------------- /tuning/tuningML/sklearndummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/tuning/tuningML/sklearndummy.py -------------------------------------------------------------------------------- /tuning/tuningML/sklearnregressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/tuning/tuningML/sklearnregressor.py -------------------------------------------------------------------------------- /utils/IOHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/utils/IOHelper.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/tables_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/utils/tables_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ardkastrati/EEGEyeNet/HEAD/utils/utils.py --------------------------------------------------------------------------------