Second Neuron in the hidden layer
fig = plt.figure(figsize=(10, 5))
ax = plt.axes(projection='3d')
bar2 = ax.scatter3D(xs=result[:, 0], ys=result[:, 1], zs=out12.reshape(-1), c= out12.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, 150)
ax.set_title('3D visualization of our neural network output layer')