Python get_input_fn示例

示例#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)

示例#2

文件： ncf_main.py 项目： 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)

示例#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)

示例#4

文件： ncf_main.py 项目： 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)