python - “AutoTrackable”对象在 Python 中不可调用

Question

我的 tensorflow 版本是 2.0

tensorflow_hub 版本是 0.7

python版本是3.7

我有这些代码

import tensorflow as tf
import tensorflow_hub as hub
import summarizer_data_utils

specials = ["<EOS>", "<SOS>","<PAD>","<UNK>"]
word2ind, ind2word,  missing_words = summarizer_data_utils.create_word_inds_dicts(words_counted,
                                                                       specials = specials)

#embed = hub.Module("https://tfhub.dev/google/nnlm-en-dim128/1")
embed = hub.load("https://tfhub.dev/google/Wiki-words-250/1")
emb = embed([key for key in word2ind.keys()])

我收到这个错误

Processing Time:  32.69666647911072
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-4-aefe0bf334ec> in <module>
     84 #embed = hub.Module("https://tfhub.dev/google/nnlm-en-dim128/1")
     85 embed = hub.load("https://tfhub.dev/google/Wiki-words-250/1")
---> 86 emb = embed([key for key in word2ind.keys()])
     87 
     88 with tf.Session() as sess:

TypeError: 'AutoTrackable' object is not callable

我该如何解决？

编辑:我找到了我的解决方案。我只需要改变

https://tfhub.dev/google/Wiki-words-250/1

到

https://tfhub.dev/google/Wiki-words-250/2

Answer 1

这是 TensorFlow 中常见问题的一部分:
https://www.tensorflow.org/hub/common_issues

解决方案是提取您需要的签名，例如:

embed = hub.load('https://tfhub.dev/google/nnlm-en-dim128/1')
embed.signatures['default'](['my text', 'batch'])