How to train a custom machine learning model using TensorFlow?