302 - remove .squeeze() from 1-D numpy array

prediction.data.numpy() is already 1-D, so the .squeeze() is unnecessary.

Author: keineahnung2345

tutorial-contents/302_classification.py

@@ -59,12 +59,12 @@ def forward(self, x):
         # plot and show learning process
         plt.cla()
         prediction = torch.max(out, 1)[1]
-        pred_y = prediction.data.numpy().squeeze()
+        pred_y = prediction.data.numpy()
         target_y = y.data.numpy()
         plt.scatter(x.data.numpy()[:, 0], x.data.numpy()[:, 1], c=pred_y, s=100, lw=0, cmap='RdYlGn')
         accuracy = float((pred_y == target_y).astype(int).sum()) / float(target_y.size)
         plt.text(1.5, -4, 'Accuracy=%.2f' % accuracy, fontdict={'size': 20, 'color':  'red'})
         plt.pause(0.1)
 
 plt.ioff()
-plt.show()
+plt.show()