Este código foi desenvolvido no vídeo "How to Make Word Vectors from Game of Thrones (LIVE)" de Siraj Raval.
Devido as constantes atualizações na biblioteca Gensim, precisamos realizar algumas intervenções no código original, conforme instruções a seguir (é possível que ainda mais atualizações aconteçam após a publicação deste repositório, portanto, caso tenha problemas de execução, acesse a documentação do Gensim para verificar a maneira correta de realizar as chamadas).
1) Substitua:
print("Word2Vec vocabulary length:", len(thrones2vec.vocab))
por
print("Word2Vec vocabulary length:", len(thrones2vec.wv.vocab))
2) Substitua:
thrones2vec.train(sentences)
por
thrones2vec.train(sentences, total_examples=thrones2vec.corpus_count, epochs=30)
3) Substitua:
all_word_vectors_matrix = thrones2vec.syn0
por
all_word_vectors_matrix = thrones2vec.wv.vectors