zhangbo пре 8 месеци
родитељ
комит
39327f8b41

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

@@ -78,10 +78,10 @@ object train_01_xgb_ad_20240808{
         map.put(fv(0), NumberUtils.toDouble(fv(1), 0.0))
       }
 
-      val v: Array[AnyRef] = new Array[AnyRef](features.length + 1)
-      v(0) = label.asInstanceOf[AnyRef]
+      val v: Array[Object] = new Array[Object](features.length + 1)
+      v(0) = label
       for (i <- 0 until features.length) {
-        v(i + 1) = map.getOrDefault(features(i), 0.0d).asInstanceOf[AnyRef]
+        v(i + 1) = map.getOrDefault(features(i), 0.0d)
       }
       RowFactory.create(v)
     })