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