Model's accuracy vs Epochs' graphs showing the results obtained using a learning rate of 0.0001, the adam optimizer, and 128 neurons in a fully connected layer, for:
- Categorical Classification for the entire dataset
- Categorical Classification for the entire dataset (to distinguish between the different failure modes and not-failed), and additionally using data augmentation - horizontal and vertical flips
- Categorical Classification just for the failed wafers (to distinguish between the different failure modes only), and using data augmentation
- Binary Classification for the entire dataset to tell whether a wafer has failed or not, also using data augmentation
- Binary Classification for equal parts of failed and non-failed wafers to tell whether a wafer has failed or not, once again using data augmentation