Reducing code duplication

[THargreaves]

Now that we have all model implementations in Python and TensorFlow, there is a considerable amount of duplicate code across notebooks. These should be consolidated together. It may also be beneficial to create modules for the TensorFlow models to store them outside the notebooks as we have done with Stan files when benchmarking.