For the classifier, we~are using exactly the same dense layer