Трудности с eval_input_receiver_fn () для модели Keras ResNet50 - PullRequest
0 голосов
/ 24 июня 2019

Я пытаюсь настроить пользовательский eval_input_receiver_fn() для передачи на tfma.export.export_eval_savedmodel() следующим образом:

model = tf.keras.models.load_model('/Users/user/Documents/../model_name.h5')
resnet50 = tf.feature_column.numeric_column('resnet50_input:0')
def receiver_fn():
    serialized_tf_example = tf.compat.v1.placeholder(
        dtype=tf.string, shape=[None], name='input_example_tensor')

    receiver_tensors = {'examples': serialized_tf_example}

    feature_spec = tf.feature_column.make_parse_example_spec(
        [resnet50])
    features = tf.io.parse_example(serialized_tf_example, feature_spec)
    return tfma.export.EvalInputReceiver(
        features=features,
        receiver_tensors=receiver_tensors)

tfma.export.export_eval_savedmodel(
    estimator=estimator,
    export_dir_base=eval_model_dir,
    eval_input_receiver_fn=receiver_fn)

Однако я получаю следующую ошибку:

TypeError: функция EvalInputReceiver должна вызываться с указанными метками

Как указать метки в моем случае?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...