碎点篇——kears fit_generator 之后获取 loss 损失的值

最新推荐文章于 2022-07-17 21:59:11 发布

万道一

最新推荐文章于 2022-07-17 21:59:11 发布

阅读量950

点赞数 1

分类专栏：碎章文章标签：知识碎点

本文链接：https://blog.csdn.net/qq_38299170/article/details/103836580

版权

在Keras中使用fit_generator训练模型后，如何获得训练和验证过程中的loss损失值？通过打印结果可以观察到loss和val_loss，分别对应训练集和验证集的损失。这有助于监测模型训练效果。

摘要由CSDN通过智能技术生成

返回主目录

kears fit_generator 之后，如何获取 loss 损失的值。

比方说，train_loss 和 val_loss 的值

model = Model([model_body.input, *y_true], model_loss)
model.fit_generator(data_generator_wrap(train, BATCH_SIZE, input_shape, anchors, num_classes),
　　　　　　　　　　　　steps_per_epoch=max(1, num_train // BATCH_SIZE),
　　　　　　　　　　　　validation_data=data_generator_wrap(val, BATCH_SIZE, input_shape, anchors, num_classes),
　　　　　　　　　　　　validation_steps=max(1, num_val // BATCH_SIZE),
　　　　　　　　　　　　epochs=1,
　　　　　　　　　　　　initial_epoch=0)

求此处的 loss

已找到方法：

h = model.fit_generator(data_generator_wrap(train, BATCH_SIZE, input_shape, 
　　　　　　　　　　　　　　anchors, num_classes),
　　　　　　　　　　　　　　steps_per_epoch=max(1, num_train // BATCH_SIZE),