|
@@ -12,7 +12,7 @@ class StaticModel():
|
|
tag_features_dict = [
|
|
tag_features_dict = [
|
|
{"vid": 5034920},
|
|
{"vid": 5034920},
|
|
{"video_style": 10382},
|
|
{"video_style": 10382},
|
|
- {"captions_color": 1082},
|
|
|
|
|
|
+ {"captions_color": 3000},
|
|
{"valid_time": 1054},
|
|
{"valid_time": 1054},
|
|
{"audience_age_group": 100},
|
|
{"audience_age_group": 100},
|
|
{"cate2": 107},
|
|
{"cate2": 107},
|
|
@@ -36,7 +36,7 @@ class StaticModel():
|
|
|
|
|
|
self.tag_feature_num = len(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]
|
|
self.tag_feature_nums = [list(d.values())[0] for d in tag_features_dict]
|
|
-
|
|
|
|
|
|
+ print(f"debug debug data tag_feature_nums: {self.tag_feature_nums}")
|
|
# 第一层的输入维度是所有特征的embedding拼接
|
|
# 第一层的输入维度是所有特征的embedding拼接
|
|
self.input_dim = self.tag_feature_num + self.stat_features_num*3
|
|
self.input_dim = self.tag_feature_num + self.stat_features_num*3
|
|
print(f"debug debug data input dim: {self.input_dim}")
|
|
print(f"debug debug data input dim: {self.input_dim}")
|