Neural Network based Decoders for the Surface Code