in hype/adjacency_matrix_dataset.pyx [0:0]
def __cinit__(self, adj, nnegs, batch_size, num_workers, burnin = False,
sample_dampening=0.75):
self.burnin = burnin
self.num_workers = num_workers
self.batch_size = batch_size
self.max_tries = 10 * nnegs
self.neg_multiplier = 1
self.queue = queue.Queue(maxsize=num_workers)
self.nnegs = nnegs
self._sample_dampening = sample_dampening
self.ids = adj['ids']
self.neighbors = adj['neighbors']
self.offsets = adj['offsets']
self.weights = adj['weights']
self.objects = adj['objects']
self.N = len(self.objects)
self._setup_alias_tables()