|
@@ -37,6 +37,7 @@ object train_01_xgb_ad_20240808{
|
|
|
val func_object = param.getOrElse("func_object", "binary:logistic")
|
|
|
val func_metric = param.getOrElse("func_metric", "auc")
|
|
|
val repartition = param.getOrElse("repartition", "20").toInt
|
|
|
+ val subsample = param.getOrElse("subsample", "0.95").toDouble
|
|
|
|
|
|
val loader = getClass.getClassLoader
|
|
|
val resourceUrl = loader.getResource(featureFile)
|
|
@@ -84,7 +85,7 @@ object train_01_xgb_ad_20240808{
|
|
|
.setMissing(0.0f)
|
|
|
.setMaxDepth(max_depth)
|
|
|
.setNumRound(num_round)
|
|
|
- .setSubsample(0.8)
|
|
|
+ .setSubsample(subsample)
|
|
|
.setColsampleBytree(0.8)
|
|
|
.setScalePosWeight(1)
|
|
|
.setObjective(func_object)
|