Handle Automatic Train/Test Split

Right now the server only supports a train/validation split with the entire dataset, and no test set is generated when the dataset is loaded.

Create a new environment variable in `docker-compose.yml` for train/test split, and then use this new test set to provide additional statistics on model training.