Проблема с загрузкой предварительно обученного Universal Sentence Encoder - PullRequest
Новая
       138

Проблема с загрузкой предварительно обученного Universal Sentence Encoder

0 голосов
/ 29 апреля 2020

У меня проблема с загрузкой предварительно обученного модуля для задачи НЛП, и проблема из-за миграции tf, я полагаю. Сайт Tensorflow сообщает, что проблема может быть решена, если переменная подписи задана правильно. Можете ли вы помочь мне исправить этот код?

Ошибка типа: объект 'AutoTrackable' не вызывается

[код] 

import tensorflow_hub as hub
# enabling the pretrained model for trainig our custom model using tensorflow hub
module_url = "https://tfhub.dev/google/universal-sentence-encoder-large/3"
embed = hub.load(module_url)

# creating a method for embedding and will using method for every input layer 
def UniversalEmbedding(x):
    return embed(tf.squeeze(tf.cast(x, tf.string)), signature='default', as_dict=True)["default"]
...