Sectionย 1. Chapterย 29
single
Challenge: Creating Word Embeddings
Swipe to show menu
Task
Swipe to start coding
You have a text corpus stored in corpus variable. Your task is to train a Word2Vec model to generate word embeddings for the given corpus. To do this:
- Import the class for creating a Word2Vec model.
- Tokenize each sentence in the
'Document'column of thecorpusby splitting each sentence into words separated by whitespaces. Store the result in thesentencesvariable. - Initialize the Word2Vec model by passing
sentencesas the first argument and setting the following parameters:- embedding size: 50;
- context window size: 2;
- minimal frequency of words to include in the model: 1;
- model: skip-gram.
- Print the top-3 most similar words to the word 'bowl'.
Solution
Everything was clear?
Thanks for your feedback!
Sectionย 1. Chapterย 29
single
Ask AI
Ask AI
Ask anything or try one of the suggested questions to begin our chat