|
@@ -15,7 +15,7 @@ type TrainReq struct {
|
|
|
// ConfigParams
|
|
|
// @Description: 训练参数
|
|
|
type ConfigParams struct {
|
|
|
- Seed float64 `json:"seed" description:"随机种子,如:42"`
|
|
|
+ Seed int `json:"seed" description:"随机种子,如:42"`
|
|
|
Objective string `json:"objective" description:"目标(回归任务),枚举值,squarederror、multi、sofamax;默认值:squarederror" `
|
|
|
EvalMetric string `json:"eval_metric" description:"评估指标,枚举值,rmse、auc logloss、merror;默认值:rmse"`
|
|
|
MaxDeltaStep int `json:"max_delta_step" description:"最大步长,如:5;正整数,必须大于0"`
|
|
@@ -23,7 +23,7 @@ type ConfigParams struct {
|
|
|
NumBoostRound int `json:"num_boost_round" description:"迭代次数;正整数,必须大于0"`
|
|
|
LearningRate float64 `json:"learning_rate" description:"学习率,如:0.0881"`
|
|
|
MaxDepth int `json:"max_depth" description:"最大深度(控制树的深度,防止过拟合),如:4;正整数,必须大于0"`
|
|
|
- MinChildWeight float64 `json:"min_child_weight" description:"最小子节点权重(防止过拟合),如:6.0601"`
|
|
|
+ MinChildWeight int `json:"min_child_weight" description:"最小子节点权重(防止过拟合),如:6.0601"`
|
|
|
Subsample float64 `json:"subsample" description:"随机采样(防止过拟合),如:0.9627"`
|
|
|
ColsampleBytree float64 `json:"colsample_bytree" description:"特征随机采样(防止过拟合),如:0.7046"`
|
|
|
Gamma float64 `json:"gamma" description:"控制分裂,如:0.4100"`
|