Browse Source

feat:添加ros回归模型

zhaohaipeng 1 month ago
parent
commit
9726e8d958

+ 2 - 2
recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/recsys_01_ros_reg_xgb_train.scala

@@ -158,10 +158,10 @@ object recsys_01_ros_reg_xgb_train {
         )
       }.map {
         case (vid, (all, label, originLabel, score, originScore)) =>
-          (vid, (all, label / all, score / all, originLabel / all, originScore / all))
+          (vid, all, label / all, score / all, originLabel / all, originScore / all)
       }
       .collect()
-      .sortBy(_._2)
+      .sortBy(-_._2)
       .take(200)
       .map(_.productIterator.mkString("\t"))
       .foreach(println)