【发布时间】:2019-12-18 12:30:10
【问题描述】:
我正在使用余弦相似度损失代码进行简单的微调,并出现以下错误。你能帮我理解问题并解决吗?
张量流版本:2.0
错误:
ValueError:找不到可以处理输入的数据适配器:(包含类型 {"class 'str'"} 的值的类 'list'),类 'numpy.ndarray'>
代码
import tensorflow as tf
import tensorflow_hub as hub
from tensorflow import keras
huburl='https://tfhub.dev/google/universal-sentence-encoder-large/5'
left_loaded_module_obj = hub.load(huburl)
left_input = keras.Input(shape=(), dtype=tf.string)
# left_trainable_embedding_layer = hub.KerasLayer(huburl,trainable=True)
left_trainable_embedding_layer = hub.KerasLayer(left_loaded_module_obj,trainable=True)
embedding_left_output= left_trainable_embedding_layer(left_input)
model = tf.keras.Model(left_input, embedding_left_output)
model.compile('sgd', loss=tf.keras.losses.CosineSimilarity(axis=-1))
model.summary()
import tensorflow_hub as hub
import tensorflow as tf
tf.enable_eager_execution()
embed = hub.load("https://tfhub.dev/google/universal-sentence-encoder-large/5")
text_list=["The quick brown fox jumps over the lazy dog.",
"I am a sentence for which I would like to get its embedding"]
embeddings = embed(text_list)
print (embeddings.numpy())
embed_target=embeddings.numpy()
print (type(embed_target),embed_target.shape)
model.fit(x=text_list,y=embed_target)
【问题讨论】:
-
检查您要输入模型的输入“类型”。
标签: python tensorflow keras tensorflow2.0 tensorflow-hub