TensorFlow模型训练中验证准确率缺失的解决方法

TensorFlow 模型训练中验证准确率缺失的解决方法

在使用 TensorFlow 进行模型训练时，您可能遇到过这样的情况：模型训练完成后，没有显示验证的准确率。

history = model.fit(train_generator, steps_per_epoch=50, epochs=5, verbose=1, validation_data=test_generator, validation_steps=50)

出现这种现象的原因可能是您在模型编译时没有设置相应的指标来计算准确率。

解决方法

在模型编译时添加 metrics=['accuracy'] 参数，即可计算验证准确率。

model.compile(optimizer='adam',
              loss='categorical_crossentropy',
              metrics=['accuracy'])

history = model.fit(train_generator,
                    steps_per_epoch=50,
                    epochs=5,
                    verbose=1,
                    validation_data=test_generator,
                    validation_steps=50)

这样，训练过程中的验证准确率就会被计算并显示。