Skip to content

embedding_outside not used in Skipgram.forward() (01 - Word2Vec (Skipgram).ipynb) #6

@MohammadElsharqawy

Description

@MohammadElsharqawy

Hi! 👋

In the notebook 01 - Word2Vec (Skipgram).ipynb, the Skipgram class defines two separate embedding layers:

self.embedding_center = nn.Embedding(voc_size, emb_size)
self.embedding_outside = nn.Embedding(voc_size, emb_size)

However, in the forward() method, only embedding_center is used — even for the outside words and vocab logits:

outside_embedding = self.embedding_center(outside)
all_vocabs_embedding = self.embedding_center(all_vocabs)

It seems this might just be a small typo. Should these lines instead use self.embedding_outside? Like this:

outside_embedding = self.embedding_outside(outside)
all_vocabs_embedding = self.embedding_outside(all_vocabs)

Thanks so much for the great notebook! Really enjoying it. 🚀

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions