예제 #1
0
 def _predict_input_fn():
     feature_map = parsing_ops.parse_example(
         input_lib.limit_epochs(serialized_examples, num_epochs=1),
         feature_spec)
     features = linear_testing_utils.queue_parsed_features(feature_map)
     features.pop('y')
     return features, None
 def _predict_input_fn():
     feature_map = tf.compat.v1.io.parse_example(
         tf.compat.v1.train.limit_epochs(serialized_examples,
                                         num_epochs=1), feature_spec)
     features = linear_testing_utils.queue_parsed_features(feature_map)
     features.pop('y')
     return features, None
예제 #3
0
 def _train_input_fn():
     feature_map = parsing_ops.parse_example(serialized_examples,
                                             feature_spec)
     features = linear_testing_utils.queue_parsed_features(feature_map)
     labels = features.pop('y')
     return features, labels