The CNN architecture used in this implementation consists of two convolutional layers, each followed by a max-pooling layer and dropout for regularisation. After the convolutional layers, the output is flattened and passed through a fully connected layer with 64 units and a dropout layer. Finally, a softmax activation function is applied to the output layer, which consists of as many units as there are classes in the dataset.
Install python, we used version 3.10.11, and the packages listed below:
- numpy
- tensorflow
- scikit-learn
- Clone the repo
git clone https://github.com/andreaslborg/OticonChallenge2023
- Install packages
pip install numpy tensorflow scikit-learn
- Run and train the model on all the data and predict on the test data
python .\CNN2D_100.py
- Run and train the model on 70% of the data and validate on 30%
python .\CNN2D_70.py