是否可以从保存的CNN中提取分配给训练图像的类别概率？

Question

我训练了CNN来执行分类任务，我将vgg16与categorical_crossentropy损失函数，sgd优化器和softmax激活函数一起使用。

我将经过训练的模型保存到h5文件中。 我知道，获得一个新的看不见的图像的类概率向量的函数是predict。

1. 但是，我想知道的是分配给训练图像的最终类别概率是否与模型一起保存？
2. 如果是，如何在不使用predict函数的情况下从保存的模型中提取此信息？

Answer 1

  

HDF5 (.h5, .hdf5)是适合存储大文件的文件格式   多维数值数组的集合（例如模型，数据   文件）。

1. 是的，您可以在训练期间将“分配给训练图像的最终类别概率” 保存到.h5文件中。

2. 但是由于您仅“将已保存的模型保存到.h5文件中 ，所以只有网络模型（如果具有权重，体系结构和优化器配置，如此定义）（更多信息，请参见here-“作为HDF5文件”部分。）

最简单的选择就是您已经提到的内容：“从保存的模型中提取此信息，而 [无需使用预测功能”。。 / p>