Как указано выше. Я вижу этот урок, который говорит о классификации, мне интересно, применима ли регрессия (я сомневаюсь в этом).
https://www.tensorflow.org/neural_structured_learning/tutorials/adversarial_keras_cnn_mnist