TFlearn IndexError超出范围

时间:2017-05-03 03:20:08

标签: tflearn index-error

我有一些看起来像这样的数据:

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”。

我认为这可能是由于指定数据的形状不正确或与Fully_connected图层有关。

这个错误是什么意思?这是因为形状错了吗?我需要在上面的代码中更改什么才能防止出现此错误?

非常感谢任何帮助。

1 个答案:

答案 0 :(得分:1)

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

List index out of......

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

tf.reset_default_graph()

tf在这里意味着tensorflow,所以不要忘记导入tensorflow

我希望这对您有帮助