Algorithm 1. CNN-AE pseudo-code | |
---|---|
Input: dataset }, training epochs N, batch size B, number of folds K | |
// Auto-encoders initialization | |
1: | Create 10 autoencoders with initial random parameters: { |
// Autoencoders training | |
2: | Partition samples in to 10 subsets: |
3: | For i = 1:10 |
4: | Train on and perform validation on |
// Augmented data generation | |
5: | |
6: | For i = 1:10 |
7: | |
8: | A |
9: | |
10: | Create CNN with initial random parameters |
11: |
// K-Fold cross validation Partition to 90% training set and 10% test set |
12: | Partition to K subsets |
13: | For k = 1:K |
14: | |
15: | |
16: | For e = 1:N |
17: | = sample_batch() |
18: | CNN.train() |
19: | = sample_batch( |
20: | CNN.validate() |
21: | CNN.test() |
22: | Return CNN |