Energy output prediction using MLPs in TensorFlow