如何使用Estimator更改默认的signature_def

Question

在signature_def中训练并提供使用DNNClassifier估计器构建的模型时，如何更改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)