Add validation and test datasets

Basic — Add a validation and test loop to avoid overfitting.

Intermediate — Learn about more complex validation and test workflows