TF - хаб-модуль универсального кодировщика предложений сохранить и перезагрузить? - PullRequest
Новая
       6

TF - хаб-модуль универсального кодировщика предложений сохранить и перезагрузить?

0 голосов
/ 20 февраля 2019

Я использую предварительно обученную модель универсального кодировщика предложений, используя следующую команду: 

import tensorflow as tf
import tensorflow_hub as hub

MODEL_NAME = 'tf-sentence-encoder'
VERSION = 1
SERVE_PATH = './models/{}/{}'.format(MODEL_NAME, VERSION)

with tf.Graph().as_default():
  module = hub.Module("https://tfhub.dev/google/universal-sentence-encoder-large/3")
  text = tf.placeholder(tf.string, [None])
  embedding = module(text)

  init_op = tf.group([tf.global_variables_initializer(),
                      tf.tables_initializer()]
                     )
  with tf.Session() as session:
      session.run(init_op)
      tf.saved_model.simple_save(session, SERVE_PATH,
                                 inputs = {"text": text}, outputs = {"embedding": embedding},
                                 legacy_init_op = tf.tables_initializer()
                                 )

Как перезагрузить сохраненную модель для API RESTFUL?

...