未添加 TF2 指标的 Keras

debugcn 发表于 Dev

传奇之尘

我在 google colab 上使用 Tensorflow 2.0 每晚构建。

我制作了简单的 CNN 模型，然后编译并拟合它。

这是代码：

model = tf.keras.models.Sequential([
    tf.keras.layers.Reshape((28, 28, 1)),
    tf.keras.layers.Conv2D(filters=32, kernel_size=(3, 3), padding='SAME',
                           activation=tf.nn.relu),
    tf.keras.layers.MaxPooling2D(pool_size=(2, 2)),
    tf.keras.layers.Conv2D(filters=64, kernel_size=(3, 3), padding='SAME',
                           activation=tf.nn.relu),
    tf.keras.layers.MaxPooling2D(pool_size=(2, 2)),
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(256, activation=tf.nn.relu),
    tf.keras.layers.Dropout(0.3),
    tf.keras.layers.Dense(10),
    tf.keras.layers.Softmax()
])


optimizer = tf.keras.optimizers.Adam(0.001)
model.compile(optimizer=optimizer,
              loss=tf.keras.losses.CategoricalCrossentropy(),
              matrics=['accuracy'])
log_dir='./logs'
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir,
                                                      histogram_freq=2,
                                                      write_images=True,
                                                      update_freq='batch',
                                                      profile_batch=0)
model.fit(x=x_train, y=y_train, batch_size=100, epochs=15,
          callbacks=[tensorboard_callback], validation_data=(x_test, y_test))

它没有给我准确的信息。

我评估了模型，它应该给我准确度信息，但它只给我损失信息。

我打印了model.metrics，它只是[]。

是bug吗？还是我错过了什么？

戴帽子的家伙

你拼错metrics了matrics。更改matrics=['accuracy']为metrics=['accuracy']。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。