2016-05-17 75 views
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,任何人都可以指出我的錯誤給出的代碼?謝謝。

+0

您需要包含完整的錯誤,而不僅僅是其中的一部分。 –

回答

3

我通過將熊貓數據框轉換爲numpy數組來實現它。