python - 加载 TensorFlow 嵌入模型

Question

我正在关注以下页面的引用:

https://tfhub.dev/google/universal-sentence-encoder/4

在代码中，模型是使用以下代码从互联网加载的:

import tensorflow as tf

embed = hub.load("https://tfhub.dev/google/universal-sentence-encoder/4")

我希望能够加载我保存在本地目录中的模型

例如:

embed = hub.Module('data\models\universal-sentence-encoder_4.tar.gz')

此代码返回以下错误

RuntimeError: Missing implementation that supports: loader

如何做到这一点？

Answer 1

问题是文件没有解压。

解压后目录路径指向解压内容所在位置，没有指定文件名。

Module 方法也更改为 load

假设解压缩的嵌入模型 .pb 文件和随附的文件夹位于指定目录中，下面的代码有效。

embed = hub.load('data\models\')