Elmo嵌入错误

Question

我正在使用ELMO为我的数据集创建嵌入，我总是得到这样的错误：

embed=elmo(data,signature="default",as_dict=True)["elmo"]
TypeError: 'AutoTrackable' object is not callable 

我的代码就像下面这样简单：

import tensorflow_hub as hub
import tensorflow as tf
elmo = hub.load("https://tfhub.dev/google/elmo/3")
embeddings = elmo(
    ["the cat is on the mat", "dogs are in the fog"],
    signature="default",
    as_dict=True)["elmo"]

Answer 1

问题出在TensorFlow版本上。ELMo不适用于tensorflow 2，请参考TensorFlow集线器文档以检查version 1和version 2中支持的文本嵌入。在将TensorFlow版本更改为1.15之后运行相同的代码就可以了。您还应该对版本1(而不是hub.load)使用hub.Module()。请参考this post了解如何更改版本。