Commit 5b1d191

authored

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

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

1 parent 906cf71 commit 5b1d191Copy full SHA for 5b1d191

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -59,12 +59,12 @@ def forward(self, x):`
`59`	`59`	`# plot and show learning process`
`60`	`60`	`plt.cla()`
`61`	`61`	`prediction = torch.max(out, 1)[1]`
`62`		`- pred_y = prediction.data.numpy().squeeze()`
	`62`	`+ pred_y = prediction.data.numpy()`
`63`	`63`	`target_y = y.data.numpy()`
`64`	`64`	`plt.scatter(x.data.numpy()[:, 0], x.data.numpy()[:, 1], c=pred_y, s=100, lw=0, cmap='RdYlGn')`
`65`	`65`	`accuracy = float((pred_y == target_y).astype(int).sum()) / float(target_y.size)`
`66`	`66`	`plt.text(1.5, -4, 'Accuracy=%.2f' % accuracy, fontdict={'size': 20, 'color': 'red'})`
`67`	`67`	`plt.pause(0.1)`
`68`	`68`
`69`	`69`	`plt.ioff()`
`70`		`-plt.show()`
	`70`	`+plt.show()`

Comments

(0)