Ejemplos de get_tfrecord_paths en Python

Ejemplo n.º 1

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,
    )

Ejemplo n.º 2

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,
    )

Ejemplo n.º 3

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,
    )