03_data_components.py 文件源码

def gen_data(num_items):
    features_array = np.arange(num_items * 3).reshape(num_items, -1)
    labels_array = np.random.choice(10, size=num_items)
    data = features_array, labels_array

    index = np.arange(num_items)
    # when your data fits into memory, just preload it
    dataset = Dataset(index=index, batch_class=MyBatch, preloaded=data)
    return dataset