|
|
@@ -76,14 +76,18 @@ object XGBoostTrain {
|
|
|
|
|
|
// 创建 XGBoostClassifier 对象
|
|
|
val xgbClassifier = new XGBoostClassifier()
|
|
|
- .setEta(0.1f)
|
|
|
+ .setEta(0.01f)
|
|
|
+ .setSubsample(0.8)
|
|
|
+ .setColsampleBytree(0.8)
|
|
|
+ .setScalePosWeight(1)
|
|
|
+ .setSeed(2024)
|
|
|
.setMissing(0.0f)
|
|
|
.setFeaturesCol("features")
|
|
|
.setLabelCol("label")
|
|
|
.setMaxDepth(5)
|
|
|
.setObjective("binary:logistic")
|
|
|
.setNthread(1)
|
|
|
- .setNumRound(5)
|
|
|
+ .setNumRound(100)
|
|
|
.setNumWorkers(1)
|
|
|
|
|
|
// 训练模型
|
