错误:使用卷积创建深度神经网络的张量流中的语法错误

时间:2019-12-04 13:58:31

标签: python tensorflow machine-learning syntax computer-vision

每次提交后,我都无法成功提交代码。由于出现以下错误文件“”,第46行     返回history.epoch,history.history ['acc'] [-1]                                                      ^ SyntaxError:在函数外部“返回”

    def train_mnist():

    class myCallback(tf.keras.callbacks.Callback):
        def on_epoch_end(self, epoch, logs={}):
            if(logs.get('acc') > 0.998):
                print("\nReached 99.8% accuracy so cancelling training!")
                self.model.stop_training = True



mnist = tf.keras.datasets.mnist
(training_images, training_labels), (test_images,
                                     test_labels) = mnist.load_data(path=path)


training_images = training_images.reshape(60000, 28, 28, 1)
test_images = test_images.reshape(10000, 28, 28, 1)

training_images = training_images/255
test_images = test_images/255

mycallback = myCallback()

model = tf.keras.models.Sequential([

    tf.keras.layers.Conv2D(32, (3, 3), activation='relu',
                           input_shape=(28, 28, 1)),
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(
        128, activation='relu'),
    tf.keras.layers.Dense(
        10, activation='softmax')

])


model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy', metrics=['accuracy'])
history = model.fit(training_images, training_labels,
                    epochs=20, callbacks=[mycallback])

 return history.epoch, history.history['acc'][-1] 

0 个答案:

没有答案