besskge.batch_sampler

Classes for sampling batches of positive and negative triples for each processing device, according to the BESS distribution scheme.

Classes

RandomShardedBatchSampler(...[, ...])	Sample random indices (with replacement) from all triple partitions.
RigidShardedBatchSampler(...[, ...])	At each call, sample triples with the same specified indices from all triple partitions, repeating triples in shorter ones to pad to the same length.
ShardedBatchSampler(partitioned_triple_set, ...)	Base class for sharded batch sampler.