Ejemplo n.º 1
0
 def input_fn_eval(params, split):
     preprocess_fn = preprocessing.get_preprocess_fn(preprocess_eval)
     return data.get_dataset(dataset_name, split, FLAGS.validation_percent,
                             params["batch_size"], preprocess_fn).repeat()
Ejemplo n.º 2
0
 def input_fn_train(params):
     preprocess_fn = preprocessing.get_preprocess_fn(preprocess)
     return data.get_dataset(dataset_name, data.DatasetSplit.TRAIN,
                             FLAGS.validation_percent, params["batch_size"],
                             preprocess_fn)