|
@@ -68,7 +68,7 @@ object train_01_xgb_ad_20250104 {
|
|
|
features,
|
|
|
negSampleRate
|
|
|
)
|
|
|
- println("path %s, train data size:%d".format(trainPath, trainData.count()))
|
|
|
+ // println("path %s, train data size:%d".format(trainPath, trainData.count()))
|
|
|
val trainDataSet: Dataset[Row] = spark.createDataFrame(trainData, schema)
|
|
|
val vectorAssembler = new VectorAssembler().setInputCols(features).setOutputCol("features")
|
|
|
val xgbInput = vectorAssembler.transform(trainDataSet).select("features", "label")
|