def run_infer(load_model, load_sess, filename, hparams, sample_num_file):
# load sample num
with open(sample_num_file, 'r') as f:
sample_num = int(f.readlines()[0].strip())
if not os.path.exists(util.RES_DIR):
os.mkdir(util.RES_DIR)
load_sess.run(load_model.iterator.initializer, feed_dict={load_model.filenames: [filename]})
preds = []
while True:
try:
step_pred = load_model.model.infer(load_sess)
preds.extend(np.reshape(step_pred, -1))
except tf.errors.OutOfRangeError:
break
preds = preds[:sample_num]
hparams.res_name = util.convert_res_name(hparams.infer_file)
# print('result name:', hparams.res_name)
with open(hparams.res_name, 'w') as out:
out.write('\n'.join(map(str, preds)))
def run_infer(load_model, load_sess, filename, hparams, sample_num_file):
# load sample num
with open(sample_num_file, 'r') as f:
sample_num = int(f.readlines()[0].strip())
if not os.path.exists(util.RES_DIR):
os.mkdir(util.RES_DIR)
load_sess.run(load_model.iterator.initializer, feed_dict={load_model.filenames: [filename]})
preds = []
while True:
try:
step_pred = load_model.model.infer(load_sess)
preds.extend(np.reshape(step_pred, -1))
except tf.errors.OutOfRangeError:
break
preds = preds[:sample_num]
hparams.res_name = util.convert_res_name(hparams.infer_file)
# print('result name:', hparams.res_name)
with open(hparams.res_name, 'w') as out:
out.write('\n'.join(map(str, preds)))