tf.contrib.estimator.export_saved_model_for_mode

Экспорт одного графа обучения/оценки/предсказания в виде SavedModel. (устаревший)

tf.contrib.estimator.export_saved_model_for_mode(
    estimator, export_dir_base, input_receiver_fn, assets_extra=None, as_text=False,
    checkpoint_path=None, mode=model_fn_lib.ModeKeys.PREDICT
)

Подробное руководство см. в Использование SavedModel с оценщиками.

Пример использования:

classifier = tf.estimator.LinearClassifier(
    feature_columns=[age, language])
classifier.train(input_fn=input_fn, steps=1000)

feature_spec = {
    'age': tf.placeholder(dtype=tf.int64),
    'language': array_ops.placeholder(dtype=tf.string)
}
label_spec = tf.placeholder(dtype=dtypes.int64)

train_rcvr_fn = tf.contrib.estimator.build_raw_supervised_input_receiver_fn(
    feature_spec, label_spec)

export_dir = tf.contrib.estimator.export_saved_model_for_mode(
    classifier,
    export_dir_base='my_model/',
    input_receiver_fn=train_rcvr_fn,
    mode=model_fn_lib.ModeKeys.TRAIN)

# export_dir is a timestamped directory with the SavedModel, which
# can be used for serving, analysis with TFMA, or directly loaded in.
with ops.Graph().as_default() as graph:
  with session.Session(graph=graph) as sess:
    loader.load(sess, [tag_constants.TRAINING], export_dir)
    weights = graph.get_tensor_by_name(''linear/linear_model/age/weights')
    ...

Этот метод является оболочкой для _export_all_saved_models и оборачивает исходную функцию input_receiver_fn в словарь для передачи в эту функцию. Подробные документы см. в _export_all_saved_models.

См. tf.contrib.estimator.export_saved_model_for_mode для текущей версии этой функции.