瀏覽代碼

dssm过滤掉无品类的item

zhangbo 4 月之前
父節點
當前提交
61a28dab25

+ 1 - 1
src/main/scala/com/aliyun/odps/spark/examples/makedata_dssm/makedata_i2i_06_itemPred_20241206.scala

@@ -132,7 +132,7 @@ object makedata_i2i_06_itemPred_20241206 {
         if (ifFilterCate){
           val cate1 = JSON.parseObject(feature).getOrDefault("category1", "无").toString
           val cate2 = JSON.parseObject(feature).getOrDefault("category2_1", "无").toString
-          !cate1.equals("无") || !cate2.equals("无")
+          !Set("无", "", "unknown").contains(cate1) || !Set("无", "", "unknown").contains(cate2)
         }else{
           true
         }