update docs

update docs

Author: 浅梦

docs/requirements.readthedocs.txt

@@ -0,0 +1 @@
+tensorflow==1.15.4

docs/source/Layers.rst

@@ -1,4 +1,4 @@
-DeepCTR Layers API
+DeepCTR-Torch Layers API
 ======================
 
 

docs/source/Models.rst

@@ -1,4 +1,4 @@
-DeepCTR-PyTorch Models API
+DeepCTR-Torch Models API
 ======================
 
 .. toctree::

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