From 4e0dd071e79439d1baa87180971659a327787b40 Mon Sep 17 00:00:00 2001 From: Koki Saitoh Date: Wed, 20 Jun 2018 08:39:46 +0900 Subject: [PATCH] Add arg for fit() method --- ch06/train_rnnlm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ch06/train_rnnlm.py b/ch06/train_rnnlm.py index 16255926..eacbf795 100644 --- a/ch06/train_rnnlm.py +++ b/ch06/train_rnnlm.py @@ -30,7 +30,8 @@ trainer = RnnlmTrainer(model, optimizer) # 勾配クリッピングを適用して学習 -trainer.fit(xs, ts, max_epoch, batch_size, time_size, max_grad) +trainer.fit(xs, ts, max_epoch, batch_size, time_size, max_grad, + eval_interval=20) trainer.plot(ylim=(0, 500)) # テストデータで評価