Not getting correct output

code ide : https://ide.codingblocks.com/s/246333

I m getting wrong ans when using pred_actor() function. Unable to figure out where the code is going wrong.

hey @Aayush ,

There is nothing as such wrong in your code , just the important thing which you need to know is that Word2Vec is highly sensitive to parameters you use to train it like min_count,window size,etc. and it also has some default randomness in its training procedure. Hence you will get different results even if you provide the same parameters again.

So to improve it and get more better results you need to play with those parameters or provide the model with more data.

Note: if you want to have a look at word vectors/embedding you just trained . Have a look at TSNE plotting technique or Tensorboard embedding projector . You can use those embeddings created using Word2Vec model and visualize them for better learning and understanding purpose.

I hope this would have resolved your doubt.
Thank You and Happy coding. :slightly_smiling_face: