我需要开发一个RNN模型,并希望使用一个数据生成器来提供训练/评估循环。
首先,在从csv文件获取数据时,我有这个帮助函数。
RECORD_DEFAULTS_TRAIN = [[0], [0.0], [0.0], [0.0], [0.0], [0.0], [0.0]]
def decode_csv(line):
parsed_line = tf.decode_csv(line, RECORD_DEFAULTS_TRAIN)
label = parsed_line[-1] # label is the last element of the list
del parsed_line[-1] # delete the last element from the list
del parsed_line[0] # even delete the first element bcz it is assumed NOT to be a feature
features = tf.stack(parsed_line) # Stack features so that you can later vectorize forward prop., etc.
return features, label
下面是我的数据生成器函数:
def data_generator(file_path_list, batch_size):
filenames = tf.placeholder(tf.string, shape=[None])
dataset = tf.data.Dataset.from_tensor_slices(filenames)
dataset = dataset.flat_map(lambda filename: tf.data.TextLineDataset(filename).skip(1).map(decode_csv))
dataset = dataset.shuffle(buffer_size=1000)
dataset = dataset.batch(batch_size)
iterator = dataset.make_initializable_iterator()
next_element = iterator.get_next()
with tf.Session() as sess:
while True:
sess.run(iterator.initializer, feed_dict={filenames: file_path_list})
while True:
try:
batch_data, batch_labels = sess.run(next_element)
# Dimension of the data needs to be: (batch_size, length_of_each_sequence, nr_inputs_in_each_timestep)
# Since the last batch in a epoch can have a different size,
# "batch_data.shape[0]" is used instead of batch_size
batch_data = np.reshape(batch_data, (batch_data.shape[0], SEQUENCE_LEN, 1))
except tf.errors.OutOfRangeError:
break
yield (batch_data, batch_labels)
以下是我如何建立和训练我的模型:
lstm_model = Sequential()
lstm_model.add(LSTM(5, input_shape=(SEQUENCE_LEN, 1), return_sequences=True))
lstm_model.add(LSTM(5, input_shape=(SEQUENCE_LEN, 1), return_sequences=False))
lstm_model.add(Dense(1))
opt = tf.keras.optimizers.Adam(lr=0.001, decay=0.0009)
lstm_model.compile(loss='mean_absolute_error', optimizer=opt, metrics=['accuracy'])
lstm_model.fit_generator(data_generator(TRAIN_FILE_PATHS, TRAIN_BATCH_SIZE), #generator,
steps_per_epoch=(NR_TRAIN_EXAMPLES // TRAIN_BATCH_SIZE),
epochs=NR_EPOCHS,
verbose=1,
validation_data=data_generator(DEV_FILE_PATHS, TRAIN_BATCH_SIZE),
validation_steps=(NR_DEV_EXAMPLES // DEV_BATCH_SIZE))
下面是我在测试集上评估模型的方法:
lstm_model.evaluate_generator(data_generator(TEST_FILE_PATHS, TEST_BATCH_SIZE),
steps=(NR_TEST_EXAMPLES // TEST_BATCH_SIZE),
verbose=1)
经过培训和评估,我在日志中看到以下错误:
IndexError: pop from empty list
以下是培训结束后日志的最后一部分:
Epoch 200/200
2/2 [==============================] - 0s 28ms/step - loss: 0.0091 - acc: 0.0120 - val_loss: 0.0128 - val_acc: 0.0000e+00
Exception ignored in: <generator object data_generator at 0x7fa97d9e34c0>
Traceback (most recent call last):
File "<ipython-input-7-2ef5e6514df7>", line 33, in data_generator
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1530, in __exit__
self._default_graph_context_manager.__exit__(exec_type, exec_value, exec_tb)
File "/usr/lib/python3.6/contextlib.py", line 99, in __exit__
self.gen.throw(type, value, traceback)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 5025, in get_controller
context.context().context_switches.pop()
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/eager/context.py", line 136, in pop
self.stack.pop()
IndexError: pop from empty list
Exception ignored in: <generator object data_generator at 0x7fa97d9e3678>
Traceback (most recent call last):
File "<ipython-input-7-2ef5e6514df7>", line 33, in data_generator
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1530, in __exit__
self._default_graph_context_manager.__exit__(exec_type, exec_value, exec_tb)
File "/usr/lib/python3.6/contextlib.py", line 99, in __exit__
self.gen.throw(type, value, traceback)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 5025, in get_controller
context.context().context_switches.pop()
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/eager/context.py", line 136, in pop
self.stack.pop()
IndexError: pop from empty list
<tensorflow.python.keras.callbacks.History at 0x7fa97d8b4828>
下面是我在运行
evaluate_generator()
之后看到的:2/2 [==============================] - 0s 28ms/step
Exception ignored in: <generator object data_generator at 0x7fa97d732af0>
Traceback (most recent call last):
File "<ipython-input-7-2ef5e6514df7>", line 33, in data_generator
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py", line 1530, in __exit__
self._default_graph_context_manager.__exit__(exec_type, exec_value, exec_tb)
File "/usr/lib/python3.6/contextlib.py", line 99, in __exit__
self.gen.throw(type, value, traceback)
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 5025, in get_controller
context.context().context_switches.pop()
File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/eager/context.py", line 136, in pop
self.stack.pop()
IndexError: pop from empty list
[0.008863004390150309, 0.0]
我感到困惑的是,为什么我在上面的每一种情况下都会看到错误消息
IndexError: pop from empty list
?正常吗?还是我做错了什么? 最佳答案
解决了的。
我想解释这个问题,而不是删除我的帖子,这样也许可以帮助其他人。。
我只给出evaluate_generator(...)
函数的示例。
这就是我调用函数的方式。。
lstm_model.evaluate_generator(data_generator(TEST_FILE_PATHS, TEST_BATCH_SIZE),
steps=(NR_TEST_EXAMPLES // TEST_BATCH_SIZE),
verbose=1)
我把它改成了:
test_data_generator = data_generator(TEST_FILE_PATHS, TEST_BATCH_SIZE)
lstm_model.evaluate_generator(test_data_generator,
steps=(NR_TEST_EXAMPLES // TEST_BATCH_SIZE),
verbose=1)
问题解决了。我在不同的地方看到了两种用法,即使在网上找到的每一个信息都不一定是真的。我也不清楚为什么在修改上面的代码时会解决这个问题。如果有人知道,我很乐意听你解释。
关于python - 具有Tensorflow Dataset API的Keras Generator-IndexError:从空列表中弹出,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/52303130/