in recommended-item-search/data_preparation.py [0:0]
def split_range(num_records, num_buckets):
interval = num_records // num_buckets
head_ids = [interval*i for i in range(num_buckets)]
tail_ids = head_ids[1:] + [num_records]
return list(zip(head_ids, tail_ids))