Сохранить модель с керасом

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 , который также имеет функцию fit с аналогичным API.

Для использования save_model и load_model вам нужна настоящая модель Keras, я думаю, это ваша classifier.В частности, экземпляр класса Model от Keras.