Exemplos de make_cloud_mlengine_request_fn em Python

Exemplo n.º 1

Arquivo: query.py Projeto: shantipriyap/tensor2tensor

def main(_):
    tf.logging.set_verbosity(tf.logging.INFO)
    validate_flags()
    usr_dir.import_usr_dir(FLAGS.t2t_usr_dir)
    problem = registry.problem(FLAGS.problem)
    hparams = tf.contrib.training.HParams(
        data_dir=os.path.expanduser(FLAGS.data_dir))
    problem.get_hparams(hparams)
    if FLAGS.cloud_mlengine_model_name:
        request_fn = serving_utils.make_cloud_mlengine_request_fn(
            credentials=GoogleCredentials.get_application_default(),
            model_name=FLAGS.cloud_mlengine_model_name,
            version=FLAGS.cloud_mlengine_model_version)
    else:
        request_fn = serving_utils.make_grpc_request_fn(
            servable_name=FLAGS.servable_name,
            server=FLAGS.server,
            timeout_secs=FLAGS.timeout_secs)
    while True:
        inputs = FLAGS.inputs_once if FLAGS.inputs_once else input(">> ")
        outputs = serving_utils.predict([inputs], problem, request_fn)
        print_str = """
Input:
{inputs}

Output:
{outputs}
    """
        print(print_str.format(inputs=inputs, outputs=outputs[0]))
        if FLAGS.inputs_once:
            break

Exemplo n.º 2

def make_request_fn():
  """Returns a request function."""
  if FLAGS.cloud_mlengine_model_name:
    request_fn = serving_utils.make_cloud_mlengine_request_fn(
        credentials=GoogleCredentials.get_application_default(),
        model_name=FLAGS.cloud_mlengine_model_name,
        version=FLAGS.cloud_mlengine_model_version)
  else:

    request_fn = serving_utils.make_grpc_request_fn(
        servable_name=FLAGS.servable_name,
        server=FLAGS.server,
        timeout_secs=FLAGS.timeout_secs)
  return request_fn

Exemplo n.º 3

Arquivo: query.py Projeto: kltony/tensor2tensor

def make_request_fn():
  """Returns a request function."""
  if FLAGS.cloud_mlengine_model_name:
    request_fn = serving_utils.make_cloud_mlengine_request_fn(
        credentials=GoogleCredentials.get_application_default(),
        model_name=FLAGS.cloud_mlengine_model_name,
        version=FLAGS.cloud_mlengine_model_version)
  else:

    request_fn = serving_utils.make_grpc_request_fn(
        servable_name=FLAGS.servable_name,
        server=FLAGS.server,
        timeout_secs=FLAGS.timeout_secs)
  return request_fn