问题描述
是否可以导出/保存/加载以前训练过的autokeras模型?我了解我可以使用以下代码来保存/加载基础的tensorflow最佳模型:
model = reg.export_model()
model.save(MODEL_FILEPATH,save_format="tf")
best_model = load_model(MODEL_FILEPATH,custom_objects=ak.CUSTOM_OBJECTS)
但是,实际上这是行不通的,因为我的数据已经由autokeras进行了拟合,因此需要进行数据准备和扩展。我认为在实际拟合之前我无法访问autokeras对输入数据(X)所做的操作,因此我无法实际使用导出的tensorflow最佳模型来预测具有未准备和未缩放功能的新样本的标签。
我在这里缺少重要的东西吗?
我还注意到在autokeras临时目录中有一些二进制文件。该目录似乎是自动生成的。有没有办法使用该目录来加载以前安装的autokeras“超级”模型?
解决方法
答案是...只需使用import pickle
-https://github.com/keras-team/autokeras/issues/1081#issuecomment-645508111:)