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 超出轴 0 的范围,大小为 1。”

我认为这要么是由于指定的数据形状不正确,要么与全连接层有关。

这个错误是什么意思?是不是因为形状不对?我需要在上面的代码中进行哪些更改以防止出现此错误?

任何帮助将不胜感激。

4

1 回答 1

1

该问题已在以下线程中进行了详细讨论。

列出索引出......

显然,在顶部添加以下代码即可解决问题:

tf.reset_default_graph()

这里的tf是tensorflow的意思,所以别忘了import tensorflow

我希望这有帮助

于 2018-06-29T07:08:29.473 回答