Code tested on following environments, other version should also work:
- linux system (ubuntu 18.04)
- python 3.6
- numpy 1.19.5
- matplotlib 3.3.4
- sklearn 0.24.2
- pytorch 1.8.1
- cuda 10.2
| Train accuracy | Test accuracy | |
|---|---|---|
| Q1 | 97.50% | 87.20% |
| Q2 | 81.82% | 79.90% |
| Q3 | 97.80% | 85.60% |
| Q4 | 95.62% | 87.50% |
| Q5 | 99.22% | 99.03% |