scala train

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

@@ -138,13 +138,13 @@ object train_01_xgb_ad_20240808{
           featureMap.put(kv_(0), kv_(1).toDouble)
         }
       })
-      val res = new ArrayBuffer[Any]()
-      res.add(label)
-      res.add(cid)
-      features.foreach(r =>{
-        res.add(featureMap.getOrElse(r, 0.0D))
-      })
-      Row(res.toArray)
+      val v: Array[Any] = new Array[Any](features.length + 2)
+      v(0) = label
+      v(1) = cid
+      for (i <- 0 until features.length) {
+        v(i + 2) = featureMap.getOrElse(r, 0.0D)
+      }
+      Row(v: _*)
     })
   }
 }