To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

Commit e1d798f1 authored by okiss's avatar okiss
Browse files

Update README.md

parent 18f692af
......@@ -13,6 +13,14 @@ Please use the command
before training the model. Please make sure that the Tensorflow version should be 2.x.
## model configuration
Please configure the config.py file correctly before running the main.py file
config['data_dir'] : indicates the directory where you stored the data
config['model'] : indicates the model you want to use, choose between 'cnn', 'eegnet', 'inception', 'xception' or'deepeye'
config['downsampled'] : True if you want to use 125 data points per second instead of 500. Default is False
config['split'] : True if you want to run a clustered version of the model, please keep it to False as the clustered version is inneficient
## deepEye3 tuning
nb_filter: [32, 64]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment