def eval_validation_input_fn(): """Evaluate on validation set.""" val_files = inputs.get_tfrecord_paths(FLAGS.validation_path) return input_fn.evaluation_input_fn( tfrecord_fpaths=val_files, batch_size=FLAGS.eval_batch_size, )
def eval_train_input_fn(): """Evaluate on train set.""" train_val_files = inputs.get_tfrecord_paths(FLAGS.train_path) return input_fn.evaluation_input_fn( tfrecord_fpaths=train_val_files, batch_size=FLAGS.eval_batch_size, )
def train_input_fn(): train_files = inputs.get_tfrecord_paths(FLAGS.train_path) return input_fn.train_input_fn( tfrecord_fpaths=train_files, batch_size=FLAGS.train_batch_size, shuffle_buffer_size=FLAGS.shuffle_buffer_size, )