Трудности с eval_input_receiver_fn() для модели Keras ResNet50
Я пытаюсь сделать на заказ 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 должна вызываться с указанными метками
Как указать метки в моем случае?