feat:添加新的训练样本生成脚本

recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/recsys_01_str_xgb_train.scala

@@ -159,9 +159,9 @@ object recsys_01_str_xgb_train {
     data.map(r => {
 
       val line: Array[String] = StringUtils.split(r, '\t')
-      val label: Int = NumberUtils.toInt(line(1))
+      val label: Int = NumberUtils.toInt(line(0))
       val map: util.Map[String, Double] = new util.HashMap[String, Double]
-      for (i <- 2 until line.length) {
+      for (i <- 1 until line.length - 1) {
         val fv: Array[String] = StringUtils.split(line(i), ':')
         map.put(fv(0), NumberUtils.toDouble(fv(1), 0.0))
       }
@@ -171,7 +171,7 @@ object recsys_01_str_xgb_train {
       for (i <- 0 until features.length) {
         v(i + 1) = map.getOrDefault(features(i), 0.0d)
       }
-      v(features.length + 1) = line(0)
+      v(features.length + 1) = line(line.length - 1)
       Row(v: _*)
     })
   }