made it use figure

latex_examples/neural-network.tex

@@ -17,6 +17,7 @@ like this neural network diagram.
 
 \def\layersep{2.5cm}
 
+\begin{figure}
 \begin{tikzpicture}[shorten >=1pt,->,draw=black!50, node distance=\layersep]
     \tikzstyle{every pin edge}=[<-,shorten <=1pt]
     \tikzstyle{neuron}=[circle,fill=black!25,minimum size=17pt,inner sep=0pt]
@@ -52,8 +53,10 @@ like this neural network diagram.
     \node[annot,above of=H-1, node distance=1cm] (hl) {Hidden layer};
     \node[annot,left of=hl] {Input layer};
     \node[annot,right of=hl] {Output layer};
-\caption{bollocks}
 \end{tikzpicture}
 % End of code
+\caption{bollocks}
+\end{figure}
+
 \end{document}