不同尺寸的模型中比较Word2Vec向量
创始人
2025-01-08 10:00:23
0次

要比较不同尺寸的Word2Vec向量模型,可以按照以下步骤进行操作:

  1. 导入所需的库:
from gensim.models import Word2Vec
import numpy as np
  1. 定义训练数据和模型参数:
sentences = [['this', 'is', 'the', 'first', 'sentence'],
            ['this', 'is', 'the', 'second', 'sentence'],
            ['yet', 'another', 'sentence'],
            ['one', 'more', 'sentence'],
            ['and', 'the', 'final', 'sentence']]

model_size1 = Word2Vec(sentences, size=100, window=5, min_count=1)
model_size2 = Word2Vec(sentences, size=200, window=5, min_count=1)
  1. 获取词汇表和对应的向量:
vocab_size1 = len(model_size1.wv.vocab)
vocab_size2 = len(model_size2.wv.vocab)

vocab1 = list(model_size1.wv.vocab.keys())
vocab2 = list(model_size2.wv.vocab.keys())

vectors1 = np.stack([model_size1.wv[word] for word in vocab1])
vectors2 = np.stack([model_size2.wv[word] for word in vocab2])
  1. 比较向量:
# 计算两个模型的平均向量
mean_vector1 = np.mean(vectors1, axis=0)
mean_vector2 = np.mean(vectors2, axis=0)

# 计算两个模型的相似度
similarity = np.dot(mean_vector1, mean_vector2) / (np.linalg.norm(mean_vector1) * np.linalg.norm(mean_vector2))

完整代码示例:

from gensim.models import Word2Vec
import numpy as np

sentences = [['this', 'is', 'the', 'first', 'sentence'],
            ['this', 'is', 'the', 'second', 'sentence'],
            ['yet', 'another', 'sentence'],
            ['one', 'more', 'sentence'],
            ['and', 'the', 'final', 'sentence']]

model_size1 = Word2Vec(sentences, size=100, window=5, min_count=1)
model_size2 = Word2Vec(sentences, size=200, window=5, min_count=1)

vocab_size1 = len(model_size1.wv.vocab)
vocab_size2 = len(model_size2.wv.vocab)

vocab1 = list(model_size1.wv.vocab.keys())
vocab2 = list(model_size2.wv.vocab.keys())

vectors1 = np.stack([model_size1.wv[word] for word in vocab1])
vectors2 = np.stack([model_size2.wv[word] for word in vocab2])

mean_vector1 = np.mean(vectors1, axis=0)
mean_vector2 = np.mean(vectors2, axis=0)

similarity = np.dot(mean_vector1, mean_vector2) / (np.linalg.norm(mean_vector1) * np.linalg.norm(mean_vector2))

print("Model 1 vocab size:", vocab_size1)
print("Model 2 vocab size:", vocab_size2)
print("Similarity between the two models:", similarity)

这样就可以比较不同尺寸的Word2Vec向量模型,并计算它们的相似度。

相关内容

热门资讯

重大通报透视器!娱家麻将透视器... 重大通报透视器!娱家麻将透视器AI脚本,娱家麻将多功能切实真的是有透视器脚本1、娱家麻将辅助软件下载...
重大消息开挂!家乡大贰开挂AI... 重大消息开挂!家乡大贰开挂AI插件,家乡大贰多功能果然有开挂插件家乡大贰脚本下载中分为三种模型:家乡...
玩家必看科普外挂!河洛杠次外挂... 玩家必看科普外挂!河洛杠次外挂AI工具,河洛杠次多功能一直是真的有外挂工具亲,关键说明,河洛杠次透视...
重大来袭开挂!广西老友玩开挂A... 重大来袭开挂!广西老友玩开挂AIapp,广西老友玩多功能竟然有开挂app1、每一步都需要思考,不同水...
科技揭秘透视挂!嘟咪互动麻将透... 科技揭秘透视挂!嘟咪互动麻将透视挂AI插件,嘟咪互动麻将多功能竟然是有透视挂插件1、玩家可以在嘟咪互...
盘点几款辅助器!微乐河北麻将辅... 盘点几款辅助器!微乐河北麻将辅助器AI工具,微乐河北麻将多功能竟然存在有辅助器工具盘点几款辅助器!微...
盘点一款透视挂!星际互联透视挂... 盘点一款透视挂!星际互联透视挂AI方法,星际互联多功能竟然是有透视挂方法1、星际互联破解器简单,星际...
推荐十款开挂!MG棋牌开挂AI... 推荐十款开挂!MG棋牌开挂AI教程,MG棋牌多功能一直是真的有开挂教程1、每一步都需要思考,不同水平...
记者揭秘辅助器!大圣天辅助器A... 记者揭秘辅助器!大圣天辅助器AI教程,大圣天多功能切实真的有辅助器教程1)大圣天免费钻石:进一步探索...
技术分享作弊器!赣牌圈打炸作弊... 技术分享作弊器!赣牌圈打炸作弊器AI插件,赣牌圈打炸多功能确实是有作弊器插件1、起透看视 赣牌圈打炸...