def split_range()

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