文件名称：SkipGram_Negative_Sampling
文件大小：6.04MB
文件格式：ZIP
更新时间：2024-04-23 04:42:46
Python
SkipGram_Negative_Sampling 带有负采样的skipgram模型的Pytorch实现。 word2vec的概念 支持的功能 跳过图 批量更新 负采样 可视化 如下所示，将嵌入内容和单词保存到TSV文件中，然后将这两个TSV文件上传到以实现更好的可视化效果。 def save_embedding(self, outdir, idx2word): embeds = self.in_embed.weight.data.cpu().numpy() f1 = open(os.path.join(outdir, 'vec.tsv'), 'w') f2 = open(os.path.join(outdir, 'word.tsv'), 'w') for idx in range(len(embeds)):
【文件预览】：
SkipGram_Negative_Sampling-main
----train.py(3KB)
----model.py(2KB)
----README.md(962B)
----data()
--------text8_toy.txt(971KB)
----preprocess.py(1KB)
----result()
--------vec.tsv(14.58MB)
--------word.tsv(79KB)