Обработка естественного языка с PyTorch - PullRequest
0 голосов
/ 11 апреля 2020

Есть ли какая-то проблема с кодом в главе 1 «Обработка естественного языка с Pytorch»? такой как Пример 1-1. Следующий код в примере 1-1: xticklabels = vocab, я не могу найти словосочетание, которое было указано в предыдущем тексте. и если запустить этот код, не могу показать ни одного изображения. Ему нравится потеря кода, импортирующая matplotlib, и он должен добавить код 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'])
...