Prepare your data for ML | Text Classification - Coding TensorFlow