add dropout layer

src/models/esmm/model.py

@@ -16,7 +16,7 @@ def build_deep_layer(net, params):
                               units=num_hidden_units,
                               activation=tf.nn.relu,
                               kernel_initializer=tf.glorot_uniform_initializer())
-        net = tf.layers.dropout(net, rate=0.1, training=True)
+        net = tf.layers.dropout(net, rate=0.3, training=True)
     return net
 
 

src/train_tractate.py

@@ -73,8 +73,7 @@ def main():
     set_essm_model_save_path("tractate", save_path)
     print("============================================================")
 
-    # TODO save
-    # save_path = get_essm_model_save_path("tractate")
+    save_path = get_essm_model_save_path("tractate")
     print("load path: " + save_path)
 
     # save_path = str(Path("~/data/models/tractate/1598236893").expanduser())  # local