Модель Keras Loaded всегда тренируется, а не предсказывает - PullRequest
0 голосов
/ 17 апреля 2020

Я занимаюсь проектом глубокого обучения. После тренировки сохраняю модель как h5. В другом файле я загружаю сохраненную модель и использую модель для прогнозирования. Однако когда я запускаю код в Pycharm, модель снова начинает тренироваться. Я перезагружаю свой ноутбук и снова запускаю, но та же самая вещь все еще появляется. Pycharm работает на неправильном файле?

model.save('model_10000.h5')

Затем в другом файле

    model = load_model('model_10000.h5')
    model.compile(loss='binary_crossentropy',
                  optimizer='adam',
                  metrics=['accuracy'])
    # predict for test set
    pred = model.predict(testX)

это то, что я получил enter image description here

Ответы [ 2 ]

0 голосов
/ 17 апреля 2020

Вам не нужно пересобирать модель. сохраненная модель всегда компилируется, и когда вы загружаете ее обратно, она всегда возвращает скомпилированную модель (см. FAQ по Keras) .

Итак, просто удалите шаг model.compile, и вы будете хорошо до go.

0 голосов
/ 17 апреля 2020

Вам не нужно компилировать сохраненную модель, возможно, это как-то связано с ней.

model.save('model_10000.h5')
model = load_model('model_10000.h5')
pred = model.predict(testX)

Проверьте это для более подробной информации: https://www.tensorflow.org/tutorials/keras/save_and_load

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