My code based is on the official TensorFlow tutorial: «Train your first neural network: basic classification»