1.可视化
from gensim.models import Word2Vec
......
class EpochLogger(CallbackAny2Vec):
'''Callback to log information about training'''
# def __init__(self):
# self.epoch = 0
# def on_epoch_end(self, model):
# loss = model.get_latest_training_loss()
# print('Loss after epoch {}: {}'.format(self.epoch, loss))
# self.epoch += 1
def __init__(self):
self.epoch = 0
self.loss_to_be_subed = 0
def on_epoch_end(self, model):
loss = model.get_latest_training_loss()
loss_now = loss - self.loss_to_be_subed
self.loss_to_be_subed = loss
print('Loss after epoch {}: {}'.format(self.epoch, loss_now))
self.epoch += 1
epoch_logger = EpochLogger()
Word2Vec( walks, size = dimensions, window = 10, min_count=0, sg=1,
workers = 8, iter = iter,callbacks=[epoch_logger])