node2vec

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])