Keras Tuner保存最佳模型

Question

我目前正在尝试使用keras调谐器为我的CNN创建一个模型，尽管我在保存我的模型以备将来使用时遇到了一些问题。

正如我所习惯的那样，我可以经常使用model.save(filename)保存我的模型来接收.model文件；但是，当尝试使用如下代码执行此操作时：

tuner = RandomSearch(
    build_model,
    objective = "val_accuracy",
    max_trials = 5,
    executions_per_trial = 1,
    directory = LOG_DIR
)

tuner.search(x= x_train, y= y_train, epochs= 1, batch_size=64, validation_data= (x_test, y_test))

bestModels = tuner.get_best_models(num_models=1)
highestScoreModel= models[0]

highestScoreModel.fit(x=x_train, y=y_train, batch_size=64, epochs=5, verbose=1, validation_split=0.2)

highestScoreModel.save("Trained_Model")

我收到了一个Trained_Model文件夹，里面没有模型，只有参数。如果有人能帮助我保存实际训练过的模型，我将非常感激。

=================编辑/更新================

我现在已经找到了一种通过筛选生成的试用文件来获得trial_id的方法。不过，当我运行时：

trial_id = getID()
tuner.save_model(trial_id=trial_id, model=highestScoreModel, step=0)

似乎什么都没有发生，没有出现保存文件。再次，我将非常感谢在这件事上的任何帮助。

Answer 1

尝尝这个

Tuner.save_model(trial_id, model, step=0)

哪里

• trial_id是你的模型试用版number
• model是你训练过的模型
• 和step是epochs numbers