deepJDOT icon indicating copy to clipboard operation
deepJDOT copied to clipboard

model.train_on_batch prevents overfitting!

Open xiangruyimoxiaoyangyang opened this issue 5 years ago • 0 comments

Hi! Thank you for the procedure, I recently applied your program to solve some practical problems. But I encountered some difficulties. In your jdot_align model, you use self.model.train_on_batch([data], [np.vstack((ys, l_dummy)), g_dummy]) to train, but I encountered the problem of overfitting. I learned by checking the information that the function train_on_batch doesn't have Callback functions, So I can't save the best model. Do you have any other good solutions? thanks!

xiangruyimoxiaoyangyang avatar Jan 06 '20 08:01 xiangruyimoxiaoyangyang