def input_fn_eval():
if flags_obj.encode_method == 'kmer':
input_fn = input_function_train_kmer(flags_obj.input_tfrec, 1,
flags_obj.batch_size,
flags_obj.cpus)
else:
input_fn = input_function_train_one_hot(flags_obj.input_tfrec, 1,
flags_obj.batch_size,
flags_obj.cpus)
return input_fn
def input_fn_eval():
if flags_obj.encode_method == 'kmer':
input_fn = input_function_train_kmer(flags_obj.input_tfrec, 1,
flags_obj.batch_size,
flags_obj.cpus)
if flags_obj.model_name in [
'embed_pool', 'embed_cnn', 'embed_lstm',
'embed_cnn_no_pool'
]:
input_fn = input_function_train_kmer_pad_to_fixed_len(
flags_obj.input_tfrec, 1, flags_obj.batch_size,
flags_obj.cpus, flags_obj.max_len, flags_obj.kmer)
else:
input_fn = input_function_train_one_hot(flags_obj.input_tfrec, 1,
flags_obj.batch_size,
flags_obj.cpus,
flags_obj.max_len)
return input_fn