The code in the chapter

[nininininini]

Does there some code issue in the chapter1? such as Example 1-1. The following code in
the example 1-1, xticklabels = vocab, I can't find the vocab was stated in the previous text.
and if run this code, can't show any image. It likes the code loss import the matplotlib and should add the code plt.show().

from sklearn.feature_extraction.text
import CountVectorizer import seaborn as sns
corpus = ['Time flies flies like an arrow.', 'Fruit flies like a banana.']
one_hot_vectorizer = CountVectorizer(binary=True)
one_hot = one_hot_vectorizer.fit_transform(corpus).toarray()
sns.heatmap(one_hot, annot=True, cbar=False, xticklabels=vocab, yticklabels=['Sentence 2'])

[jiangzhonglian]

谢谢提醒，可以帮忙提交一下 pull request 一起来完善吗？

[ferquintana84]

any response? same problem here

[yu-cheng-kuo-28]

Hi there, I reckon this is the only way:

=> BOOK: NLP with PyTorch

from sklearn.feature_extraction.text import CountVectorizer
import seaborn as sns

corpus = ['Time flies flies like an arrow.', 'Fruit flies like an banana.']
print(one_hot_vectorizer.vocabulary_)
=> {'time': 6, 'flies': 3, 'like': 5, 'an': 0, 'arrow': 1, 'fruit': 4, 'banana': 2}

print(sorted(one_hot_vectorizer.vocabulary_))
=> ['an', 'arrow', 'banana', 'flies', 'fruit', 'like', 'time']
vocab = sorted(one_hot_vectorizer.vocabulary_)

one_hot_vectorizer = CountVectorizer(binary=True)
one_hot = one_hot_vectorizer.fit_transform(corpus).toarray()

sns.heatmap(one_hot, annot=True, cbar=False, xticklabels= vocab, yticklabels= ['Sentence_1','Sentence_2']);

BTW, you can also do things like this to pick up the words you want

vocab_2 = sorted(one_hot_vectorizer.vocabulary_)[1:5]
sns.heatmap(one_hot[:,1:5], annot=True, cbar=False, xticklabels= vocab_2, yticklabels= ['Sentence_1','Sentence_2']);