|
@@ -76,12 +76,12 @@ 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)
|
|
|
+ val v: Array[Any] = new Array[Any](features.length + 1)
|
|
|
v(0) = label
|
|
|
for (i <- 0 until features.length) {
|
|
|
v(i + 1) = map.getOrDefault(features(i), 0.0d)
|
|
|
}
|
|
|
- RowFactory.create(v: _*)
|
|
|
+ Row(v: _*)
|
|
|
})
|
|
|
}
|
|
|
}
|