fixed example dataset

damian0604 · Apr 22, 2020 · ee64b73 · ee64b73
1 parent 114cfc4
commit ee64b73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rm-course-2/week11/lecture11.pdf b/rm-course-2/week11/lecture11.pdf
diff --git a/rm-course-2/week11/lecture11.tex b/rm-course-2/week11/lecture11.tex
@@ -244,8 +244,8 @@ \subsection{Principal Component Analysis and Singular Value Decomposition}
 import matplotlib.pyplot as plt
 %matplotlib inline
 
-autotexts = datasets.fetch_20newsgroups('rec.autos', remove=('headers', 'footers', 'quotes'), subset='train')['data']
-religiontexts = datasets.fetch_20newsgroups('soc.religion.christian', remove=('headers', 'footers', 'quotes'), subset='train')['data']
+autotexts = datasets.fetch_20newsgroups(categories=['rec.autos'], remove=('headers', 'footers', 'quotes'), subset='train')['data']
+religiontexts = datasets.fetch_20newsgroups(categories=['soc.religion.christian'], remove=('headers', 'footers', 'quotes'), subset='train')['data']
 
 texts = autotexts[:20] + religiontexts[:20]
 \end{lstlisting}