2016-05-17 14 views
1

私はマルチクラス分類のためのニューラルネットワークモデルに合うようにしようとしていたされているが、私はのpythonは、ニューラルネットワークを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の配列に変換することで動作させました。

関連する問題