How to write a training loop

1. Prepare a dataset

2. Create a Iterator

3. Define a model

4. Select a Optimizer

5. Write a training loop

6. Save the trained model

7. Make a prediction using the trained model