Ejemplo n.º 1
0
def _internal_eval(model, global_step, sess, iterator, iterator_feed_dict,
                   summary_writer, label):
    """Computing perplexity."""
    sess.run(iterator.initializer, feed_dict=iterator_feed_dict)
    ppl = model_helper.compute_perplexity(model, sess, label)
    utils.add_summary(summary_writer, global_step, "%s_ppl" % label, ppl)
    return ppl
Ejemplo n.º 2
0
def _internal_eval(model, global_step, sess, real_iterator, iterator_feed_dict,
                   iterator_handle, summary_writer, label):
  """Computing perplexity."""
  sess.run(real_iterator.initializer, feed_dict=iterator_feed_dict)
  ppl, all_summaries = model_helper.compute_perplexity(model, sess, label,
                                                       iterator_handle)

  utils.add_summary(summary_writer, global_step, "%s_ppl" % label, ppl)
  for key in all_summaries:
    utils.add_summary(summary_writer, global_step, key, all_summaries[key])
  return ppl