我正在尝试拟合文本分类模型。因此我想使用 tensorflow-hub 提供的 text_embedding_column 函数。不幸的是我得到一个运行时错误
import tensorflow_hub as hub
embedded_text_feature_column = hub.text_embedding_column(
key="sentence",
module_spec="https://tfhub.dev/google/nnlm-en-dim128/1")
我得到的错误如下:
RuntimeError Traceback (most recent call last)
<ipython-input-18-df9239a27166> in <module>()
2 embedded_text_feature_column = hub.text_embedding_column(
3 key="sentence",
----> 4 module_spec="https://tfhub.dev/google/nnlm-en-dim128/1")
/anaconda3/lib/python3.6/site-packages/tensorflow_hub/feature_column.py in text_embedding_column(key, module_spec, trainable)
72 ValueError: if module_spec is not suitable for use in this feature column.
73
---> 74 module_spec = module.as_module_spec(module_spec)
75 _check_module_is_text_embedding(module_spec)
76 return _TextEmbeddingColumn(key=key, module_spec=module_spec,
/anaconda3/lib/python3.6/site-packages/tensorflow_hub/module.py in as_module_spec(spec)
31 return spec
32 elif isinstance(spec, six.string_types):
---> 33 return load_module_spec(spec)
34 else:
35 raise ValueError("Unknown module spec type: %r" % type(spec))
/anaconda3/lib/python3.6/site-packages/tensorflow_hub/module.py in load_module_spec(path)
56
57 path = registry.resolver(path)
---> 58 return registry.loader(path)
59
60
/anaconda3/lib/python3.6/site-packages/tensorflow_hub/registry.py in __call__(self, *args, **kwargs)
43 raise RuntimeError(
44 "Missing implementation that supports: %s(*%r, **%r)" % (
---> 45 self._name, args, kwargs))
46
47
RuntimeError: Missing implementation that supports: loader(*('/var/folders/pc/h0fr0z2x1pjbmdb63mhn84_w0000gn/T/tfhub_modules/32f2b2259e1cc8ca58c876921748361283e73997',), **{})