def on_epoch_end(self, model):
print("Epoch #{} end, used time: {}min".format(
self.epoch, (time.time() - self.time) / 60))
print('loss: {}'.format(
Word2Vec.get_latest_training_loss(model) - self.loss_save))
self.epoch += 1
self.loss_save = Word2Vec.get_latest_training_loss(model)
def on_epoch_end(self, model: Word2Vec):
loss = model.get_latest_training_loss() - self.prev_loss
historic_loss.append(loss)
epochs_list.append(self.epoch)
print('Loss after epoch {}: {}'.format(self.epoch, loss))
self.prev_loss += loss
self.epoch += 1
