Как продолжить сохранить историю потери проверки в Керасе - PullRequest
Новая
       10

Как продолжить сохранить историю потери проверки в Керасе

0 голосов
/ 05 января 2019

Допустим, я хочу тренироваться со 100 эпохой. Но после 50 эпох я хотел бы сохранять модель раз в 5 эпох. Я также хотел бы сохранить историю. Я не был уверен, как это сделать, поэтому я сделал что-то вроде ниже. Пожалуйста, дайте мне знать, как продолжить сохранять историю. 

history = model.fit(X_train, y_train, validation_split = 0.2, shuffle=True, batch_size=gb.BATCH_SIZE, epochs=50)

counters = gb.EPOCHS_COUNT  // 5

for counter in range(counters):
    history += model.fit(X_train, y_train, validation_split = 0.2, shuffle=True, batch_size=gb.BATCH_SIZE, epochs=5)
    modelName = "model_weights_total_0"+str(counter)+".h5"
    model.save(model_save_path+modelName, overwrite=True)

1 Ответ

0 голосов
/ 05 января 2019

Одним из способов является суммирование history.history значений вручную. 

val_losses = []
for counter in range(counters):
    history = model.fit(X_train, y_train, validation_split = 0.2, shuffle=True, batch_size=gb.BATCH_SIZE, epochs=5)
    val_losses += history.history['val_loss']
...