Third Neuron in the hidden layer

In [30]:
fig = plt.figure(figsize=(10, 5))
ax = plt.axes(projection='3d')
bar3 = ax.scatter3D(xs=result[:, 0], ys=result[:, 1], zs=out13.reshape(-1), c= out13.reshape(-1), cmap=cm.coolwarm);
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')

# we use view_init in changing the viewing angle
ax.view_init(50, 250)

ax.set_title('3D visualization of our neural network output layer')
Out[30]:
Text(0.5,0.92,'3D visualization of our neural network output layer')