3

次のようなデータがあります。

X = [[1,2,3,4],[01010],[-1.6]]
y = [[4,2]]

tflearn を使用して、このデータでニューラル ネットワークをトレーニングしようとしています。データの形状を変更したことを除いて、TFlearn github ホームページ ( https://github.com/tflearn/tflearn ) にあるのと同じ例を使用しています。

tflearn.init_graph(num_cores=1)

net = tflearn.input_data(shape=[None, 2,2,1])
net = tflearn.fully_connected(net, 64)
net = tflearn.dropout(net, 0.5)
net = tflearn.fully_connected(net, 10, activation='softmax')
net = tflearn.regression(net, optimizer='adam', loss='categorical_crossentropy')

model = tflearn.DNN(net)
model.fit(X,y)

このエラーが発生し続けます:

「IndexError: インデックス 2 は、サイズ 1 の軸 0 の範囲外です。」

これは、指定されているデータの形状が正しくないか、完全に接続されたレイヤーに関係しているためだと思います。

このエラーはどういう意味ですか? 形が悪いからでしょうか?このエラーを回避するには、上記のコードで何を変更する必要がありますか?

どんな助けでも大歓迎です。

4

1 に答える 1