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

Question

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

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

best_model.txt

分类器

模块

temp_model

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

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

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模型（而不是将其转换为其他模型），则这是一种合适的解决方案。