blob: f100db7b38671c2012646ae0a4466c654957cbca [file] [log] [blame]
op {
name: "BoostedTreesCalculateBestGainsPerFeature"
input_arg {
name: "node_id_range"
type: DT_INT32
}
input_arg {
name: "stats_summary_list"
type: DT_FLOAT
number_attr: "num_features"
}
input_arg {
name: "l1"
type: DT_FLOAT
}
input_arg {
name: "l2"
type: DT_FLOAT
}
input_arg {
name: "tree_complexity"
type: DT_FLOAT
}
input_arg {
name: "min_node_weight"
type: DT_FLOAT
}
output_arg {
name: "node_ids_list"
type: DT_INT32
number_attr: "num_features"
}
output_arg {
name: "gains_list"
type: DT_FLOAT
number_attr: "num_features"
}
output_arg {
name: "thresholds_list"
type: DT_INT32
number_attr: "num_features"
}
output_arg {
name: "left_node_contribs_list"
type: DT_FLOAT
number_attr: "num_features"
}
output_arg {
name: "right_node_contribs_list"
type: DT_FLOAT
number_attr: "num_features"
}
attr {
name: "max_splits"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "num_features"
type: "int"
has_minimum: true
minimum: 1
}
}