1
我試圖適應多類分類神經網絡模型,但我有蟒蛇keras神經網絡:IndexError:指數是出界外
IndexError: indices are out-of-bounds
錯誤。 我的訓練數據的維度爲(26728, 450)
,有450個功能。輸出大小爲5(5類)。我用to_categorical(train_Y)
將它轉換爲5列的矩陣。
的代碼是
model = Sequential()
model.add(Dense(64, input_dim=train_X.shape[1], init='uniform'))
model.add(Activation('tanh'))
model.add(Dropout(0.5))
model.add(Dense(64, init='uniform'))
model.add(Activation('tanh'))
model.add(Dropout(0.5))
model.add(Dense(5, init='uniform'))
model.add(Activation('softmax'))
model.compile(
loss='categorical_crossentropy',
optimizer='sgd',
metrics=['accuracy']
)
train_Y_keras = to_categorical(train_Y)
model.fit(train_X, train_Y_keras, nb_epoch=10)
我不完全瞭解加層的,我複製和修改,這裏http://keras.io/getting-started/functional-api-guide/#getting-started-with-the-keras-functional-api,任何人都可以指出我的錯誤給出的代碼?謝謝。
您需要包含完整的錯誤,而不僅僅是其中的一部分。 –