Implement logging for the model

[felixlut]

Currently we only print the accuracy and loss of the model, it would be beneficial if we also implemented a system for logging it, so that we can see which parameters (batch_size, seed, epochs, etc) yields what result (accuracy, val_accuracy, etc)