Linear regression using TensorFlow