updates for documentation

data-workspaces · Jan 5, 2020 · 9b58506 · 9b58506
1 parent e45f408
commit 9b58506
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/dataworkspaces/kits/tensorflow.py b/dataworkspaces/kits/tensorflow.py
@@ -22,7 +22,12 @@
     from dataworkspaces.kits.tensorflow1 import add_lineage_to_keras_model_class
     
     # Wrap our model class. This is the only DWS-specific change needed.
-    keras.Sequential = add_lineage_to_keras_model_class(keras.Sequential)
+    # We add an optional checkpoint configuration, which will cause checkpoints
+    # to be written to the workspace's scratch directory and then the best
+    # checkpoint copied to the results resource.
+    keras.Sequential = add_lineage_to_keras_model_class(keras.Sequential,
+                           checkpoint_config=CheckpointConfig(model='fashion',
+                                                              monitor='loss'))
     
     fashion_mnist = keras.datasets.fashion_mnist
     

diff --git a/docs/kits.rst b/docs/kits.rst
@@ -144,5 +144,5 @@ TensorFlow
 
 .. automodule:: dataworkspaces.kits.tensorflow
    :no-undoc-members:
-   :members: add_lineage_to_keras_model_class
+   :members: DwsModelCheckpoint,CheckpointConfig,add_lineage_to_keras_model_class