Keras 生成器

Question

我写了我的生成器，但在输出中出现错误

def batch_generator(X_data, y_data, batch_size):
    samples_per_epoch = X_data.shape[0]
    number_of_batches = samples_per_epoch/batch_size
    counter=0
    index = np.arange(np.shape(y_data)[0])
    while 1:
        index_batch = index[batch_size*counter:batch_size*(counter+1)]
        X_batch = X_data[index_batch,:].toarray()
        y_batch = y_data[y_data.index[index_batch]]
        counter += 1
        yield X_batch,y_batch
        if (counter > number_of_batches):
            counter=0

在线错误：y_batch = y_data[y_data.index[index_batch]]

'numpy.ndarray' object has no attribute 'index'

Answer 1

您应该将数组转换为list 。 由于index()是list类型的方法，而不是numpy.ndarray 。 请尝试：

y_data.tolist().index(index_batch)