update docs

shenweichen · Oct 18, 2020 · bc881dc · bc881dc
1 parent 2090756
commit bc881dc
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 2 deletions.
diff --git a/docs/requirements.readthedocs.txt b/docs/requirements.readthedocs.txt
@@ -0,0 +1 @@
+tensorflow==1.15.4
diff --git a/docs/source/Layers.rst b/docs/source/Layers.rst
@@ -1,4 +1,4 @@
-DeepCTR Layers API
+DeepCTR-Torch Layers API
 ======================
 
 

diff --git a/docs/source/Models.rst b/docs/source/Models.rst
@@ -1,4 +1,4 @@
-DeepCTR-PyTorch Models API
+DeepCTR-Torch Models API
 ======================
 
 .. toctree::

diff --git a/tests/models/AFM_test.py b/tests/models/AFM_test.py
@@ -22,11 +22,20 @@ def test_AFM(use_attention, sparse_feature_num, dense_feature_num):
     check_model(model, model_name, x, y)
 
     early_stopping = EarlyStopping(monitor='val_binary_crossentropy', min_delta=0, verbose=1, patience=0, mode='min')
+
+    # test callbacks
     model_checkpoint = ModelCheckpoint(filepath='model.ckpt', monitor='val_binary_crossentropy', verbose=1,
                                        save_best_only=True,
                                        save_weights_only=False, mode='max', period=1)
     model.fit(x, y, batch_size=64, epochs=3, validation_split=0.5, callbacks=[early_stopping, model_checkpoint])
 
+    model_checkpoint = ModelCheckpoint(filepath='model.ckpt', monitor='val_binary_crossentropy', verbose=1,
+                                       save_best_only=False,
+                                       save_weights_only=False, mode='max', period=1)
+
+    model.fit(x, y, batch_size=64, epochs=3, validation_split=0.5, callbacks=[early_stopping, model_checkpoint])
+
+
 
 if __name__ == '__main__':
     pass