Блог

Главная - Вопросы по программированию - Как изменить signature_def по умолчанию с помощью оценщика

Как изменить signature_def по умолчанию с помощью оценщика

#python #tensorflow #tensorflow-serving #tensorflow-estimator

#python #тензорный поток #тензорный поток-обслуживание #tensorflow-оценщик

Вопрос:

Как мне изменить signature_def ключ при обучении и обслуживании модели, построенной с использованием DNNClassifier Estimator в Tensorflow ?

Используя saved_model_cli инструмент, я вижу, что значение по умолчанию равно predict . Я хотел бы изменить это.

Я сохраняю модель, которая будет обслуживаться следующей функцией:

   def save_serving_model(self,estimator):
      ''' 
      To run a quick check:
      saved_model_cli run --dir /path/to/mode/ --tag_set serve --signature_def predict --input_exprs="Headline=['some text'];Description=['some more text']"
      '''
      feature_placeholder = {'Headline': tf.placeholder('string', [1], name='headline_placeholder'),
        'Description': tf.placeholder('string', [1],name='description_placeholder')}
      serving_input_fn = tf.estimator.export.build_raw_serving_input_receiver_fn(feature_placeholder)
      estimator.export_savedmodel(self.serving_model_name, serving_input_fn)