Learn to generalize explicitly by learning error vector on testset batch

Assuming you installed Python 3 with tensorflow, scikit-learn and numpy you need to:

• Clone the project

git clone https://github.com/PaulEmmanuelSotir/generalization_loss.git
cd ./explicit_generalization_training

• Train the model on CIFAR dataset

python train_dnn.py

• Apply explicit generalization training on trained model to learn errors on testset predictions

python generalization_training.py