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 | Объект Tensor типа resource. Дескриптор переменной ансамбля. |
feature_ids | Список как минимум из 1 объекта Tensor типа int32. Тензор ранга 1 с идентификаторами каждого признака. Это фактический идентификатор признака, который будет использоваться при разбиении. |
dimension_ids | Список объектов Tensor типа int32. Список тензоров ранга 1, представляющих размерность каждого признака. |
node_ids | Список объектов Tensor типа int32 той же длины, что и dimension_ids. Список тензоров ранга 1, представляющих узлы, в которых для этого признака выполняется разбиение. |
gains | Список объектов Tensor типа float32 той же длины, что и dimension_ids. Список тензоров ранга 1, представляющих приросты для каждого разбиения признака. |
thresholds | Список объектов Tensor типа int32 той же длины, что и dimension_ids. Список тензоров ранга 1, представляющих пороговые значения для каждого разбиения признака. |
left_node_contribs | Список объектов Tensor типа float32 той же длины, что и dimension_ids. Список тензоров ранга 2 с вкладами левых листьев для каждого разбиения признака. Они будут добавлены к значениям предыдущих узлов, чтобы получить значения левых узлов. |
right_node_contribs | Список объектов Tensor типа float32 той же длины, что и dimension_ids. Список тензоров ранга 2 с вкладами правых листьев для каждого разбиения признака. Они будут добавлены к значениям предыдущих узлов, чтобы получить значения правых узлов. |
split_types | Список объектов Tensor типа string той же длины, что и dimension_ids. Список тензоров ранга 1, представляющих тип разбиения для каждого признака. |
max_depth | Объект Tensor типа int32. Максимальная глубина строящегося дерева. |
learning_rate | Объект Tensor типа float32. Коэффициент сжатия для каждого нового дерева. |
pruning_mode | Объект Tensor типа int32. 0 — без отсечения, 1 — предварительное отсечение, 2 — последующее отсечение. |
logits_dimension | Необязательный int. По умолчанию — 1. Скаляр, размерность логитов. |
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/api_docs/python/tf/raw_ops/BoostedTreesUpdateEnsembleV2