tfhub简介

最新推荐文章于 2022-12-24 17:00:00 发布

*Major*

最新推荐文章于 2022-12-24 17:00:00 发布

阅读量1.1k

点赞数

本文链接：https://blog.csdn.net/qq_41375318/article/details/104265378

版权

$t f h u b 简介$

import tensorflow_hub as hub

embed = hub.load("./nnlm-zh-dim50-with-normalization_2")

embeddings = embed(["男 人", "女 人"])
embeddings.shape

在这里插入图片描述

embeddings[0]

在这里插入图片描述

embeddings[1]

在这里插入图片描述

v1 = embeddings[0].numpy().reshape(1,-1)
v2 = embeddings[1].numpy().reshape(1,-1)

import sklearn
sklearn.metrics.pairwise.cosine_similarity(v1, v2)

在这里插入图片描述

import tensorflow as tf

hub_layer = hub.KerasLayer("./nnlm-zh-dim50-with-normalization_2", output_shape=[50],
                           input_shape=[], dtype=tf.string)

model = tf.keras.Sequential()
model.add(hub_layer)
model.add(tf.keras.layers.Dense(16, activation='relu'))
model.add(tf.keras.layers.Dense(1, activation='sigmoid'))

model.summary()