Python get_input_fn Examples

Example #1

 def get_pred_input_fn():
     return movielens_dataset.get_input_fn(
         False,
         distribution_utils.per_device_batch_size(FLAGS.batch_size,
                                                  num_gpus), ncf_dataset,
         FLAGS.data_dir, FLAGS.dataset, 1)

Example #2

File: ncf_main.py Project: AlwaysTheBeginer/models

 def get_pred_input_fn():
   return movielens_dataset.get_input_fn(
       False,
       distribution_utils.per_device_batch_size(FLAGS.batch_size, num_gpus),
       ncf_dataset, FLAGS.data_dir, FLAGS.dataset, 1)

Example #3

 def get_train_input_fn():
     return movielens_dataset.get_input_fn(
         True,
         distribution_utils.per_device_batch_size(FLAGS.batch_size,
                                                  num_gpus), ncf_dataset,
         FLAGS.data_dir, FLAGS.dataset, FLAGS.epochs_between_evals)

Example #4

File: ncf_main.py Project: AlwaysTheBeginer/models

 def get_train_input_fn():
   return movielens_dataset.get_input_fn(
       True,
       distribution_utils.per_device_batch_size(FLAGS.batch_size, num_gpus),
       ncf_dataset, FLAGS.data_dir, FLAGS.dataset, FLAGS.epochs_between_evals)