Я пытаюсь использовать скрипты обработки естественного языка TensorFlow для выполнения семантического поиска в PDF. Я хотел бы кэшировать локальные вложения, потому что они занимают очень много времени для создания. Я надеялся сделать это в файле protobuf, но я открою другие рекомендации. Как бы я поступил так?
Вот как выглядит код. Вложения создаются из серии предложений, которые я уже инициализировал.
tf_sentences = tf.reshape(sentences[0:1000], [-1])
embeddings = embed(
tf_sentences,
signature="default",
as_dict=True)["default"]
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
sess.run(tf.tables_initializer())
x = sess.run(embeddings)