|
@@ -189,7 +189,7 @@ object recsys_01_ros_reg_weight_xgb_train {
|
|
v(i + 1) = map.getOrDefault(features(i), 0.0d)
|
|
v(i + 1) = map.getOrDefault(features(i), 0.0d)
|
|
}
|
|
}
|
|
v(features.length + 1) = logKey
|
|
v(features.length + 1) = logKey
|
|
- v(features.length + 2) = if (label == 0) 1 else 3.0 // 设置权重
|
|
|
|
|
|
+ v(features.length + 2) = if (label == 0) 1.1 else 3.0 // 设置权重
|
|
Row(v: _*)
|
|
Row(v: _*)
|
|
})
|
|
})
|
|
}
|
|
}
|