文件名称：pytorch-sgns:PyTorch中的Skipgram负采样
文件大小：7KB
文件格式：ZIP
更新时间：2024-05-24 04:34:53
word2vec pytorch skipgram Python
PyTorch SGNS Python中Word2Vec的SkipGramNegativeSampling 。 在实现了另一个但非常普遍的。 它可以与任何嵌入方案一起使用！ 我敢打赌。 vocab_size = 20000 word2vec = Word2Vec ( vocab_size = vocab_size , embedding_size = 300 ) sgns = SGNS ( embedding = word2vec , vocab_size = vocab_size , n_negs = 20 ) optim = Adam ( sgns . parameters ()) for batch , ( iword , owords ) in enumerate ( dataloader ): loss = sgns ( iword , owords )
【文件预览】：
pytorch-sgns-master
----train.py(4KB)
----model.py(3KB)
----LICENSE(1KB)
----README.md(916B)
----.gitignore(1KB)
----preprocess.py(4KB)
----plot.py(2KB)