update tdm

models/recall/tdm/config.yaml

@@ -71,4 +71,4 @@ train:
     inference:
       dirname: "inference"
       epoch_interval: 4
-      save_last: True
\ No newline at end of file
+      save_last: True

models/recall/tdm/model.py

@@ -148,7 +148,7 @@ class Model(ModelBase):
         # 最后的概率判别FC，将所有层次的node分类结果放到一起以相同的标准进行判别
         # 考虑到树极大可能不平衡，有些item不在最后一层，所以需要这样的机制保证每个item都有机会被召回
         tdm_fc = fluid.layers.fc(input=layer_classifier_res,
-                                 size=self.label_nums,
+                                 size=2,
                                  act=None,
                                  num_flatten_dims=2,
                                  param_attr=fluid.ParamAttr(
@@ -373,7 +373,7 @@ class Model(ModelBase):
 
             # 过最终的判别分类器
             tdm_fc = fluid.layers.fc(input=layer_classifier_res,
-                                     size=self.label_nums,
+                                     size=2,
                                      act=None,
                                      num_flatten_dims=2,
                                      param_attr=fluid.ParamAttr(

models/recall/tdm/tree/layer_list.txt

+1,2
+3,4,5,6
+7,8,9,10,11,12,13
+14,15,16,17,18,19,20,21,22,23,24,25
\ No newline at end of file