使用临时保存的二进制文件加载 autokeras 模型

Question

从this post 可以看到，保存和加载 autokeras 模型存在问题（请参阅提供的链接了解更多详细信息）。我注意到 autokeras 临时文件夹中有一些二进制文件（例如 /tmp/autokeras_ECWBT7），其中包含一些二进制文件：

图（1.graph，2.graph，...）

best_model.txt

分类器

模块

临时模型

我找不到任何文档来查看如何使用它来加载经过训练的模型（如果有的话）。

如果你能帮助我解决这个问题，我将不胜感激。

Answer 1

Autokeras 确实缺乏文档。 在 ImageClassifier 的情况下，我发现了一些如何进一步使用训练模型的方法（对于 autokeras 版本 0.3.7）：

1. 如果您正在使用autokeras.image.image_supervised.ImageClassifier 并希望使用fit 继续搜索模型，或者重新启动final_fit 以在graph_directory 中找到最佳模型，您可以使用：

clf = ImageClassifier(verbose=True, augment=False, path='./path/to/graph/directory', resume=True)

1. 现在您可以继续网络架构搜索或导出您的最佳模型。如果您想导出您的最佳模型ImageClassifier 并将其用于预测，您可以使用： clf.export_autokeras_model('your_model.pkl')

2. 以后你可以这样加载它：

from autokeras.utils import pickle_from_file

model = pickle_from_file('your_model.pkl')

之后，您可以通过以下方式进行预测： y_predict = model.predict(x_test)

如果您想稍后完全使用 Autokeras ImageClassifier 模型（而不是将其转换为其他东西），这是合适的解决方案。