Update project to run with Tensorflow 2

[italocampos]

This project runs with Tensorflow 1.x, which is deprecated and the support to this version will become poor soon. In order to get this project reusable through the scientific community, we should to update the code base to run to an updated version of Tensorflow.

Below we have a message from Google Colab when running a code with Tensorflow 1.x.:

WARNING: Tensorflow 1 is deprecated, and support will be removed on August 1, 2022.
After that, `%tensorflow_version 1.x` will throw an error.

Your notebook should be updated to use Tensorflow 2.
See the guide at https://www.tensorflow.org/guide/migrate#migrate-from-tensorflow-1x-to-tensorflow-2.

[italocampos]

I'm trying to set the project dependencies to fetch tensorflow 1.x version, but I have no push permissions to the repository. I can't create PRs. :(

[oliversheridanmethven]

It can readily be set to fetch TensorFlow version 1 routines typically by using import tensorflow.compat.v1 as tf, although I do agree that a nice re-write in TensorFlow version 2 would be a good addition.

[oliversheridanmethven]

There is the very related project, also in Tensor Flow, but using version 2: https://github.com/frankhan91/DeepBSDE