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 объекта типа Tensor с типом 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 | Необязательный 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