axes.set_xlabel(f"number of neurons in {target_layer}") axes.set_ylabel("neurons in output layer") plt.show()