Ejecuta múltiples predictores de conjunto de regresión aditiva en instancias de entrada y
calcula la actualización de los registros almacenados en caché. Está diseñado para ser utilizado durante el entrenamiento. Recorre los árboles a partir de la identificación del árbol en caché y la identificación del nodo en caché y calcula las actualizaciones que se enviarán a la memoria caché.
Métodos públicos
BoostedTreesTrainingPredict estático | |
Salida <Entero> | ID de nodo () Tensor de rango 1 que contiene nuevos ID de nodo en los nuevos tree_ids. |
Salida <Flotante> | logits parciales () Tensor de rango 2 que contiene actualización de logits (con respecto a los valores en caché almacenados) para cada ejemplo. |
Salida <Entero> | identificadores de árbol () Tensor de rango 1 que contiene nuevos identificadores de árbol para cada ejemplo. |
Métodos Heredados
Métodos públicos
BoostedTreesTrainingPredict público estático crear ( Alcance alcance, Operando <?> treeEnsembleHandle, Operando <Integer> cachedTreeIds, Operando <Integer> cachedNodeIds, Iterable< Operand <Integer>> bucketizedFeatures, Long logitsDimension)
Método de fábrica para crear una clase que envuelve una nueva operación BoostedTreesTrainingPredict.
Parámetros
alcance | alcance actual |
---|---|
CachedTreeIds | Tensor de rango 1 que contiene identificadores de árbol almacenados en caché, que es el árbol inicial de predicción. |
cachedNodeIds | Tensor de rango 1 que contiene la identificación del nodo en caché, que es el nodo inicial de la predicción. |
características de cubo | Una lista de tensores de rango 1 que contienen la identificación del depósito para cada característica. |
dimensión logits | escalar, dimensión de los logits, que se utilizará para la forma de logits parciales. |
Devoluciones
- una nueva instancia de BoostedTreesTrainingPredict
Salida pública <Entero> nodeIds ()
Tensor de rango 1 que contiene nuevos ID de nodo en los nuevos tree_ids.
salida pública <flotante> logits parciales ()
Tensor de rango 2 que contiene actualización de logits (con respecto a los valores en caché almacenados) para cada ejemplo.
public Output <Integer> treeIds ()
Tensor de rango 1 que contiene nuevos identificadores de árbol para cada ejemplo.