|
@@ -76,12 +76,6 @@ class StaticModel():
|
|
|
|
|
|
|
|
def net(self, input, is_infer=False):
|
|
def net(self, input, is_infer=False):
|
|
|
# 创建模型实例
|
|
# 创建模型实例
|
|
|
- def __init__(self, static_feature_nums=[5,5,5,5,5],
|
|
|
|
|
- stat_features_num=47,
|
|
|
|
|
- stat_features_num_embeddings=100,
|
|
|
|
|
- embedding_dim=8, output_dim=16,
|
|
|
|
|
- hidden_layers=[40, 32], hidden_acts=["relu", "relu"]):
|
|
|
|
|
-
|
|
|
|
|
dssm_model = DSSMLayer(
|
|
dssm_model = DSSMLayer(
|
|
|
tag_feature_nums=self.feature_nums,
|
|
tag_feature_nums=self.feature_nums,
|
|
|
stat_features_num = self.stat_features_num,
|
|
stat_features_num = self.stat_features_num,
|
often