AI开发平台ModelArts-梯度提升树回归特征重要性:参数说明

时间：2023-11-01 16:20:20

AI开发平台ModelArts

参数	子参数	参数说明
input_columns_str	-	数据集的特征列名组成的格式化字符串，例如: "column_a" "column_a,column_b"
label_col	-	目标列名
model_input_features_col	-	特征向量的列名
prediction_col	-	训练模型时，预测结果对应的列名，默认为"prediction"
max_depth	-	树的最大深度，默认为5
max_bins	-	特征分裂时的最大分箱个数，默认为32
min_instances_per_node	-	决策树分裂时要求每个节点必须包含的实例数目，默认为1
min_info_gain	-	最小信息增益，默认为0
subsampling_rate	-	训练每棵树时，对训练集的抽样率，默认为1
max_iter	-	最大迭代次数，默认为20
step_size	-	步长，默认为0.1