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