Python serving_input_fnの例

コード例 #1

ファイル: trainer_utils.py プロジェクト: yucoian/tensor2tensor

def make_export_strategy(problem, hparams):
    return tf.contrib.learn.make_export_strategy(
        lambda: data_reader.serving_input_fn(problem, hparams), as_text=True)

コード例 #2

ファイル: trainer_utils.py プロジェクト: zeyu-h/tensor2tensor

def make_export_strategy(problem, hparams):
  return tf.contrib.learn.make_export_strategy(
      lambda: data_reader.serving_input_fn(problem, hparams), as_text=True)