File size: 858 Bytes
96ad6cc 5dc6582 96ad6cc 136e2bd 2d29cc5 136e2bd 2d29cc5 136e2bd 2d29cc5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | # Simple CNN for knotted x unknotted recognition
The model is trained on downsampled [knots_AF](https://huggingface.co/datasets/EvaKlimentova/knots_AF) dataset where sequences with length < 500 are padded with 'X's.
The architecture is [PENGUINN](https://www.frontiersin.org/articles/10.3389/fgene.2020.568546/full) with modified input.
## Model versions
(names in format `cnn_{NUMBER_OF_EPOCHS}epochs_{TEST_ACCURACY}.h5`)
**Trained on v1 version of the dataset (it contained duplicites):**
- [`cnn_10epochs_95.h5`](./cnn_10epochs_95.h5)
**Trained on v2 version of the dataset (some unknotted labels were incorrect):**
- [`cnn_5epochs_94.h5`](./cnn_5epochs_94.h5)
- [`cnn_10epochs_94.h5`](./cnn_10epochs_94.h5)
**Trained on the latest version of the dataset:**
- [`cnn_10epochs_latest.h5`](./cnn_10epochs_latest.h5) - accuracy on test: `~95.17 %` |