|
@@ -34,7 +34,7 @@ class StaticModel():
|
|
|
self.tag_feature_nums = [list(d.values())[0] for d in tag_features_dict]
|
|
self.tag_feature_nums = [list(d.values())[0] for d in tag_features_dict]
|
|
|
|
|
|
|
|
# 第一层的输入维度是所有特征的embedding拼接
|
|
# 第一层的输入维度是所有特征的embedding拼接
|
|
|
- self.input_dim = self.tag_feature_num * self.embedding_dim + self.stat_features_num*3*self.embedding_dim
|
|
|
|
|
|
|
+ self.input_dim = self.tag_feature_num + self.stat_features_num*3
|
|
|
|
|
|
|
|
|
|
|
|
|
def _init_hyper_parameters(self):
|
|
def _init_hyper_parameters(self):
|
often