Change the number of the output layer

[cherifsid]

Hello im a novice in machine learning and capsule neural network , i want to make this model train on [A-Z] optical character not the handwrtting mnist dataset , so i i create my owne dataset i have 28 classes , so my question in witch part of code i found this option , thank u