使用model.save()可以保存经过训练的模型。但是，在使用模型进行预测时，我仍然需要恢复各自的类/标签映射(0: 'cat', 1: 'dog' ... etc.)。目前，我正在从.class_indices中保存train_generator，并重新加载它以准备测试数据以完成此任务。但是，这非常不方便，因为它迫使我将映射文件保存在某个地方，以备将来使用保存的model.h5文件。
因此，我想知道是否存在一种更简单的方法来保存类信息，即将其保存在模型文件中。我在keras文档中找不到关于此的任何信息，只有这篇文章Attaching class labels to a Keras model中有人试图提出“解决方法”，但我认为必须有更好的方法。