Skip to content

Commit

Permalink
Change link to CBOW implementation
Browse files Browse the repository at this point in the history
  • Loading branch information
veekaybee committed Oct 2, 2023
1 parent d51b6c0 commit 8680d42
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion embeddings.tex
Original file line number Diff line number Diff line change
Expand Up @@ -1757,7 +1757,7 @@ \subsection{Word2Vec}
torch.save(self.model.state_dict(), self.model_path)
logging.warning(f'Save model to {self.model_path}')
\end{minted}
\caption Word2Vec CBOW implementation in PyTorch\\ \href{https://github.com/veekaybee/viberary/tree/main/word2vec}{ see full implementation here}
\caption Word2Vec CBOW implementation in PyTorch\\ \href{https://github.com/veekaybee/viberary/blob/main/src/notebooks/cbow.ipynb}{ see full implementation here}
\end{figure}
Once we've completed our iteration through our training set, we have learned a model that retrieves both the probability of a given word being the correct word, and the entire embedding space for our vocabulary.
Expand Down

0 comments on commit 8680d42

Please sign in to comment.