Keras Training Tests

These tests require an additional python dependency on sklearn, which can be installed as follows:

python3 -m pip install sklearn

These tests are not checked by the OSS CI.