Browse Source

scala train

zhangbo 8 months ago
parent
commit
733235de88

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

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