2016-04-26 2 views
2

Я использую Keras для прогнозирования временных рядов. В качестве стандарта я использую 20 эпох. Я хочу знать, что предсказала моя нейронная сеть для каждой из 20 эпох.Python/Keras - Как получить доступ к каждому прогнозу эпохи?

Используя model.predict Я получаю только одно предсказание среди всех эпох (не уверен, как Keras выбирает его). Я хочу всех предсказаний или, по крайней мере, 10 лучших.

Кто-нибудь знает, как мне помочь?

ответ

5

Я думаю, что здесь есть путаница.

Эпоха используется только при обучении нейронной сети, поэтому, когда тренировка останавливается (в данном случае, после 20-й эпохи), весы соответствуют тем, которые были рассчитаны в последнюю эпоху.

Keras распечатывает текущие значения потерь при проверке, установленной во время обучения после каждой эпохи. Если веса после каждой эпохи не сохраняются, они теряются. Вы можете сохранить весы для каждой эпохи с помощью обратного вызова ModelCheckpoint, а затем загрузить их с помощью load_weights на вашу модель.

Вы можете вычислить свои предсказания после каждой учебной эпохи путем внедрения соответствующего обратного вызова с помощью подклассов Callback и вызова предсказать на модели внутри функций on_epoch_end.

Затем, чтобы использовать его, вы создаете экземпляр обратного вызова, составляете список и используете его как ответные запросы аргументов ключевого слова до model.fit.

+0

Hello Matias! Если бы вы могли взглянуть на это, было бы весьма полезно: http://stackoverflow.com/questions/36895627/python-keras-creating-a-callback-with-one-prediction-for-each-epoch – abutremutante

Смежные вопросы