tf.raw_ops.BoostedTreesTrainingPredict
Выполняет предсказания нескольких моделей регрессии с добавлением ансамблей на входных примерах и
tf.raw_ops.BoostedTreesTrainingPredict(
tree_ensemble_handle, cached_tree_ids, cached_node_ids, bucketized_features,
logits_dimension, name=None
)
вычисляет обновление к кэшированным логам. Он предназначен для использования во время обучения. Он проходит по деревьям, начиная с кэшированного идентификатора дерева и кэшированного идентификатора узла, и вычисляет обновления, которые должны быть внесены в кэш.
| Аргументы | |
|---|---|
tree_ensemble_handle | Tensor типа resource. |
cached_tree_ids | Tensor типа int32. Одномерный тензор, содержащий кэшированные идентификаторы деревьев, которые являются начальным деревом предсказания. |
cached_node_ids | Tensor типа int32. Одномерный тензор, содержащий кэшированный идентификатор узла, который является начальным узлом предсказания. |
bucketized_features | Список из как минимум 1 Tensor объекта типа int32. Список одномерных тензоров, содержащих идентификаторы корзины для каждого признака. |
logits_dimension | int. Скалярная величина, размерность логов, используемая для частичной формы логов. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж Tensor объектов (partial_logits, tree_ids, node_ids). | |
partial_logits | Tensor типа float32. |
tree_ids | Tensor типа int32. |
node_ids | Tensor типа int32. |
© 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.4/api_docs/python/tf/raw_ops/BoostedTreesTrainingPredict