Tensorflow keras - 如果模型不存在，如何避免在加载 h5 模型时出错

Question

当我尝试像这样获取模型时，我正在编写一个专门训练机器学习模型的应用程序：

model = tf.keras.models.load_model('./models/model.h5')

我收到一个错误：

Unable to open file (unable to open file: name = 'models/model.h5', errno = 2, error message = 'No such file or directory', flags = 0, o_flags = 0)

但在某些特殊情况下，模型可能不存在于磁盘上，此时应创建、训练并保存模型以供以后使用。检查模型是否存在的正确方法是什么？我可以使用 python 中的内置功能来检查文件是否存在，但对我来说很明显 load_model 上应该有一个参数，如果文件不存在则返回 None 而不是抛出错误。

Answer 1

检查文件是否存在的 Python 方法是正确的方法。

这可能是个人的，但应该返回 None 并不明显。打开文件时，该文件必须存在。

你可以：

import os.path

if os.path.isfile(fname):
    model=load_model(fname)
else:
    model = createAndTrainModel()

或者你可以

try:
    model=load_model(fname)
except:
    model = createAndTrainModel()

我更喜欢第一个。