为了账号安全,请及时绑定邮箱和手机立即绑定

IndexError:在 TensorFlow 中列出超出范围的索引

IndexError:在 TensorFlow 中列出超出范围的索引

ABOUTYOU 2021-08-24 16:27:25
我收到一个错误,IndexError: list index out of range.Traceback 说Run id: P0W5X0Log directory: /tmp/tflearn_logs/Exception in thread Thread-2:Traceback (most recent call last):  File "/usr/local/Cellar/python@2/2.7.15/Frameworks/Python.framework/Versions/2.7/lib/python2.7/threading.py", line 801, in __bootstrap_inner    self.run()  File "/usr/local/Cellar/python@2/2.7.15/Frameworks/Python.framework/Versions/2.7/lib/python2.7/threading.py", line 754, in run    self.__target(*self.__args, **self.__kwargs)  File "/Users/xxx/anaconda/xxx/lib/python2.7/site-packages/tflearn/data_flow.py", line 201, in fill_batch_ids_queue    ids = self.next_batch_ids()  File "/Users/xxx/anaconda/xxx/lib/python2.7/site-packages/tflearn/data_flow.py", line 215, in next_batch_ids    batch_start, batch_end = self.batches[self.batch_index]IndexError: list index out of range我写了代码,# coding: utf-8import tensorflow as tfimport tflearnfrom tflearn.layers.core import input_data,dropout,fully_connectedfrom tflearn.layers.conv import conv_2d, max_pool_2dfrom tflearn.layers.normalization import local_response_normalizationfrom tflearn.layers.estimator import regressiontf.reset_default_graph()net = input_data(shape=[None,20000, 4, 42])net = conv_2d(net, 4, 16, activation='relu')net = max_pool_2d(net, 1)net = tflearn.activations.relu(net)net = dropout(net, 0.5)net = tflearn.fully_connected(net, 2, activation='softmax')net = tflearn.regression(net, optimizer='adam', learning_rate=0.5, loss='categorical_crossentropy')model = tflearn.DNN(net)model.fit(np.array(trainDataSet).reshape(1,20000, 4, 42), np.array(trainLabel), n_epoch=400, batch_size=32, validation_set=0.1, show_metric=True)pred = np.array(model.predict(np.array(testDataSet).reshape(1,20000, 4, 42)).argmax(axis=1))我真的不明白为什么会发生这样的错误。我试图改写成model.fit(np.array(trainDataSet).reshape(1,20000, 4, 42), np.array(trainLabel), n_epoch=400, batch_size=1, validation_set=0.1, show_metric=True) 因为 bach 导致了这个错误,但发生了同样的错误。我在这部分重写了另一个数字,但发生了同样的错误。我的代码有什么问题?我应该如何解决这个问题?
查看完整描述

2 回答

  • 2 回答
  • 0 关注
  • 346 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信