02_basic.py 文件源码

def gen_data(num_items):
    index = np.arange(num_items)
    data = np.arange(num_items * 3).reshape(num_items, -1)
    # when your data fits into memory, just preload it
    dataset = Dataset(index=index, batch_class=ArrayBatch, preloaded=data)
    return dataset