tf.raw_ops.BoostedTreesUpdateEnsembleV2
Обновляет ансамбль деревьев, добавляя слой к последнему растущему дереву
tf.raw_ops.BoostedTreesUpdateEnsembleV2(
tree_ensemble_handle, feature_ids, dimension_ids, node_ids, gains, thresholds,
left_node_contribs, right_node_contribs, split_types, max_depth, learning_rate,
pruning_mode, logits_dimension=1, name=None
)
или начав новое дерево.
| Аргументы | |
|---|---|
tree_ensemble_handle | Переменная типа resource. Обработчик переменной ансамбля. |
feature_ids | Список из не менее 1 объекта типа int32. Массив ранга 1 с идентификаторами для каждого признака. Это фактический идентификатор признака, который будет использоваться в разбиении. |
dimension_ids | Список объектов типа int32. Список массивов ранга 1, представляющих размерность в каждом признаке. |
node_ids | Список с длиной, равной dimension_ids, объектов типа int32. Список массивов ранга 1, представляющих узлы, для которых у этого признака есть разбиение. |
gains | Список с длиной, равной dimension_ids, объектов типа float32. Список массивов ранга 1, представляющих приросты для каждого разбиения признака. |
thresholds | Список с длиной, равной dimension_ids, объектов типа int32. Список массивов ранга 1, представляющих пороговые значения для каждого разбиения признака. |
left_node_contribs | Список с длиной, равной dimension_ids, объектов типа float32. Список массивов ранга 2 с левыми вкладами листа для каждого разбиения признака. Будут добавлены к предыдущим значениям узла, чтобы составить значения левых узлов. |
right_node_contribs | Список с длиной, равной dimension_ids, объектов типа float32. Список массивов ранга 2 с правыми вкладами листа для каждого разбиения признака. Будут добавлены к предыдущим значениям узла, чтобы составить значения правых узлов. |
split_types | Список с длиной, равной dimension_ids, объектов типа string. Список массивов ранга 1, представляющих тип разбиения для каждого признака. |
max_depth | Переменная типа int32. Максимальная глубина дерева для построения. |
learning_rate | Переменная типа float32. Коэффициент сжатия для каждого нового дерева. |
pruning_mode | Переменная типа int32. 0 — без обрезки, 1 — предварительная обрезка, 2 — пост-обрезка. |
logits_dimension | Необязательный параметр. По умолчанию 1. Скаляр, размерность логарифмов |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/BoostedTreesUpdateEnsembleV2