decode_parse_nn.py 文件源码

def decode():
  """ Decode file sentence-by-sentence  """
  with tf.Session(config=tf.ConfigProto(intra_op_parallelism_threads=NUM_THREADS)) as sess:
    # Create model and load parameters.
    with tf.variable_scope("model", reuse=None):
      model_dev, steps_done = create_model_default(sess, forward_only=True, attention=attention, model_path=model_path)
    #for v in tf.all_variables():
    #  print(v.name, v.get_shape())

#    eval_batch_size = 64
#    start_time = time.time()
#    do_evalb(model_dev, sess, dev_set, eval_batch_size)
#    time_elapsed = time.time() - start_time
#    print("Batched evalb time: ", time_elapsed)

    start_time = time.time()
    write_decode(model_dev, sess, dev_set) 
    time_elapsed = time.time() - start_time
    print("Decoding all dev time: ", time_elapsed)