Keras сохраняют модель и загружают модель, изменяя форму прогнозов - PullRequest
0 голосов
/ 15 мая 2018

Я хочу сохранить keras.wrappers.scikit_learn.KerasClassifier.Я использовал метод model.save, как рекомендует Keras, но когда я загружаю свою модель, модель keras.models.Sequential вместо keras.wrappers.scikit_learn.KerasClassifier.

Это проблема, потому что когда я хочу сделать прогноз, у меня естьвывод с этой формой:

array([[0.21577413, 0.37268943, 0.13626885, 0.2752676 ],
       [0.22569732, 0.34428614, 0.15705906, 0.2729574 ],
       [0.24492404, 0.27987108, 0.21357943, 0.2616254 ],
       [0.24831992, 0.26592442, 0.22779736, 0.25795835],
       [0.24938227, 0.26125908, 0.23273101, 0.25662765]], dtype=float32)

Вместо этого вывода, который я хочу

array([3, 3, 3, 3, 3], dtype=int64)

Итак, мой вопрос, как я могу сохранить и загрузить keras.wrappers.scikit_learn.KerasClassifier?Может быть, есть еще одна проблема, которая меняет форму вывода моих прогнозов?

Спасибо

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...