changed visualise function to closely match notebook (#626)

blei-lab · Apr 28, 2017 · 068f1c4 · 068f1c4
1 parent 52fef60
commit 068f1c4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/tex/tutorials/supervised-regression.tex b/docs/tex/tutorials/supervised-regression.tex
@@ -132,10 +132,10 @@ \subsubsection{Criticism}
 
 \begin{lstlisting}[language=Python]
 def visualise(X_data, y_data, w, b, n_samples=10):
-  w_samples = w.sample(n_samples).eval()
+  w_samples = w.sample(n_samples)[0].eval()
   b_samples = b.sample(n_samples).eval()
-  plt.scatter(X_data, y_data)
-  inputs = np.linspace(-1, 10, num=400)
+  plt.scatter(X_data[:, 0], y_data)
+  inputs = np.linspace(-8, 8, num=400)
   for ns in range(n_samples):
     output = inputs * w_samples[ns] + b_samples[ns]
     plt.plot(inputs, output)