KeyError: «Имя 'boosted_trees / QuantileAccumulator /' относится к операции, отсутствующей на графике."при загрузке сохраненной модели - PullRequest
Новая
       9

KeyError: «Имя 'boosted_trees / QuantileAccumulator /' относится к операции, отсутствующей на графике."при загрузке сохраненной модели

0 голосов
/ 04 марта 2019

Я создал оценщик TensorFlow: 

outlier_estimator = tf.estimator.BoostedTreesClassifier(
    n_batches_per_layer = 15,
    feature_columns=outlier_feature_columns, 
    model_dir="./tensorboard_logs/wifi_outliers/", 
    n_classes=2
)

и сохранил его: 

def serving_input_receiver_fn():
    inputs = {
        "signal_0": tf.placeholder(shape=[1], dtype=tf.float32, name="signal_0"),
        "signal_1": tf.placeholder(shape=[1], dtype=tf.float32, name="signal_1")
    }
    return tf.estimator.export.ServingInputReceiver(inputs, inputs)

outlier_estimator.export_savedmodel(export_dir_base="./export/", serving_input_receiver_fn=serving_input_receiver_fn)

Но при попытке загрузить сохраненную модель 

tf.reset_default_graph()
with tf.Session() as sess:
    tf.saved_model.loader.load(
        sess,
        [tf.saved_model.tag_constants.SERVING],
        "./export/1551699998"
    )

Я столкнулся с ошибкой:

KeyError: «Имя 'boosted_trees / QuantileAccumulator /' относится к операции, не отображаемой на графике."

Что такоеЯ делаю не так?

Я использую: Python 3.7 тензор потока 1.13.1

...