|
@@ -32,8 +32,7 @@ class StaticModel():
|
|
|
|
|
|
self.feature_num = len(tag_features_dict)
|
|
|
self.feature_nums = [list(d.values())[0] for d in tag_features_dict]
|
|
|
- self.embedding_dim = embedding_dim
|
|
|
- self.output_dim = output_dim
|
|
|
+
|
|
|
# 第一层的输入维度是所有特征的embedding拼接
|
|
|
self.input_dim = feature_num * embedding_dim + self.stat_features_num*3*embedding_dim
|
|
|
|