pipeline_operations.py 文件源码

def gen_data(num_items):
    ix = np.arange(num_items).astype('str')
    data = np.arange(num_items * 3).reshape(num_items, -1)
    ds = Dataset(index=ix, batch_class=MyBatch, preloaded=data)
    return ds, data


# Create datasets