使用Keras保存模型

Question

我使用方法save_model和load_mode，但是它不起作用。

我有一个错误：AttributeError：'GridSearchCV'对象没有属性'get_config' 我不知道我是否正确使用此方法。我以示例为例显示代码：

gridSearch = GridSearchCV(estimator = classifier,
                          param_grid = parameters,
                          scoring = "accuracy",
                          cv = 10)
gridSearch.fit(X_train, y_train)
save_model(gridSearch, filepath = 'monModele.h5')

结果是错误属性Error。您能帮我找到解决此问题的方法还是找到其他方法来保存和加载Keras模型吗？

Answer 1

这是因为GridSearchCV不是Keras模型，而是sklearn中的一个模块，该模块还具有带有类似API的fit函数。

要使用save_model和load_model，您需要实际的Keras模型，我猜是它是您的classifier。具体来说，是Keras的Model类的实例。