change model dim

recommend-model-produce/src/main/python/models/dssm/net.py

@@ -5,7 +5,7 @@ import numpy as np
 
 class DSSMLayer(nn.Layer):
     def __init__(self, feature_nums=[5,5,5,5,5], embedding_dim=8, output_dim=16, 
-                 hidden_layers=[64, 32], hidden_acts=["relu", "relu"]):
+                 hidden_layers=[40, 32], hidden_acts=["relu", "relu"]):
         super(DSSMLayer, self).__init__()
         
         self.feature_num = len(feature_nums)

recommend-model-produce/src/main/python/models/dssm/static_model.py

@@ -13,7 +13,7 @@ class StaticModel():
         self.feature_nums = self.config.get("hyper_parameters.feature_nums", [5,5,5,5,5])
         self.embedding_dim = self.config.get("hyper_parameters.embedding_dim", 8)
         self.output_dim = self.config.get("hyper_parameters.output_dim", 16)
-        self.hidden_layers = self.config.get("hyper_parameters.hidden_layers", [64, 32])
+        self.hidden_layers = self.config.get("hyper_parameters.hidden_layers", [40, 32])
         self.hidden_acts = self.config.get("hyper_parameters.hidden_acts", ["relu", "relu"])
         self.learning_rate = self.config.get("hyper_parameters.optimizer.learning_rate", 0.001)
         self.margin = self.config.get("hyper_parameters.margin", 0.3)  # 用于损失函数的margin参数