AI开发平台ModelArts-决策树分类:参数说明

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

AI开发平台ModelArts

参数	子参数	参数说明
b_use_default_encoder	-	是否使用默认编码，默认为True
input_features_str	-	输入的特征列名以逗号分隔组成的格式化字符串，例如: "column_a" "column_a,column_b"
label_col	-	目标列
classifier_label_index_col	-	目标列经过标签编码后的新的列名，默认为"label_index"
classifier_feature_vector_col	-	算子输入的特征向量列的列名，默认为"model_features"
prediction_index_col	-	算子输出的预测label对应的标签列，默认为"prediction_index"
prediction_col	-	算子输出的预测label的列名，默认为"prediction"
max_depth	-	树的最大深度，默认为5
max_bins	-	最大分箱数，默认为32
min_instances_per_node	-	树节点分割时要求子节点包含的最小实例数，默认为1
min_info_gain	-	最小信息增益，默认为0
impurity	-	不纯度，支持entropy、gini，默认为"gini"