AutoKeras模型的导出与导入

网上之前用export_autokeras_model()现在好像用不了了。

官方已经优化了他的导出：

model = clf.export_model()

try:
    model.save("model_autokeras", save_format="tf")
except:
    model.save("model_autokeras.h5")

from tensorflow.keras.models import load_model

loaded_model = load_model("model_autokeras", custom_objects=ak.CUSTOM_OBJECTS)

predicted_y = loaded_model.predict(tf.expand_dims(x_test, -1))
print(predicted_y)

You can easily export your model the best model found by AutoKeras as a Keras Model.

官网说这个直接导出的就是keras模型，不像之前大家需要费很大的力气。这位博主@Kanny广小隶真的是做了非常详细的解释。但现在都不用啦～～

附上官网tutorial

祝好