def batch_dev_step(x_batch, y_batch, writer=None):
"""
Evaluates model on a dev set
"""
batches = data_helpers.batch_iter(
zip(x_batch, y_batch), FLAGS.batch_size, 1)
for batch in batches:
x_batch, y_batch = zip(*batch)
feed_dict = {
cnn.input_x: x_batch,
cnn.input_y: y_batch,
cnn.dropout_keep_prob: 1.0
}
step, summaries, loss, accuracy = sess.run(
[global_step, dev_summary_op, cnn.loss, cnn.accuracy],
feed_dict)
time_str = datetime.datetime.now().isoformat()
print("{}: step {}, loss {:g}, acc {:g}".format(time_str, step, loss, accuracy))
if writer:
writer.add_summary(summaries, step)
# Generate batches
评论列表
文章目录