Обнаружение ключевой ошибки при построении простой нейронной сети с прямой связью в Керасе.

Question

Вот снимок моего набора данных, включая его форму:

Теперь вот код, который я использую для построения NN:

# define the architecture of the network
model = Sequential()
model.add(Dense(50, input_dim=X_train.shape[1], init="uniform", activation="relu"))
model.add(Dense(38, activation="relu", kernel_initializer="uniform"))
model.add(Dense(1, activation = 'sigmoid'))

print("[INFO] compiling model...")
adam = Adam(lr=0.01)
model.compile(loss="binary_crossentropy", optimizer=adam,
    metrics=["accuracy"])
model.fit(X_train, Y_train, epochs=50, batch_size=128,
    verbose=1)

Когда я это делаю, я получаю следующую ошибку:

KeyError: '[233946 164308 296688 166151 276165  88219 117980 163503 182033 164328\n 188083  30380  37984 245771 308534   6215 181186 307488 172375  60446\n  29397 166681   5587 243263 103579 262182 107823 234790 258973 116433\n 199283  86118 172148 257334 286452 248407  81280 ...] not in index'

Мне не удалось найти решение этой проблемы. Любая помощь приветствуется.

Answer 1

Я считаю, что входные данные не являются массивом numpy, как описано в этом выпуске github на странице keras

Попробуйте подобрать модель, используя это:

model.fit(np.array(X_train), np.array(Y_train), epochs=50, batch_size=128,
    verbose=1)

Что будет приводить массивы в виде пустых массивов при подборе данных.