|
@@ -144,7 +144,8 @@ object recsys_01_ros_reg_xgb_train {
|
|
|
map.put(fv(0), NumberUtils.toDouble(fv(1), 0.0))
|
|
|
}
|
|
|
|
|
|
- val v: Array[Any] = new Array[Any](features.length + 1)
|
|
|
+ val v: Array[Any] = new Array[Any](features.length + 2)
|
|
|
+ v(0) = label
|
|
|
for (i <- 0 until features.length) {
|
|
|
v(i + 1) = map.getOrDefault(features(i), 0.0d)
|
|
|
}
|