Ejecuta múltiples predictores de conjuntos de regresión aditiva en instancias de entrada y
calcula la actualización de los logits almacenados en caché. Está diseñado para usarse durante el entrenamiento. Atraviesa 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 caché.
Métodos públicos
estática BoostedTreesTrainingPredict | |
Salida <Integer> | nodeIds () Tensor de rango 1 que contiene nuevos ID de nodo en los nuevos tree_ids. |
Salida <Float> | partialLogits () Rank 2 Tensor que contiene la actualización de logits (con respecto a los valores almacenados en caché) para cada ejemplo. |
Salida <Integer> | treeIds () Tensor de rango 1 que contiene nuevos identificadores de árbol para cada ejemplo. |
Métodos heredados
Métodos públicos
public static BoostedTreesTrainingPredict crear ( Alcance alcance, operando <?> treeEnsembleHandle, operando <entero> cachedTreeIds, operando <entero> cachedNodeIds, Iterable < operando <entero >> bucketizedFeatures, Long logitsDimension)
Método de fábrica para crear una clase que envuelva una nueva operación BoostedTreesTrainingPredict.
Parámetros
alcance | alcance actual |
---|---|
cachedTreeIds | Rango 1 Tensor que contiene los identificadores de árbol en caché, que es el árbol inicial de predicción. |
cachedNodeIds | Rango 1 Tensor que contiene la identificación del nodo en caché, que es el nodo inicial de la predicción. |
bucketizedFeatures | Una lista de tensores de rango 1 que contienen la identificación del depósito para cada función. |
logitsDimension | escalar, dimensión de los logits, que se utilizará para la forma de logits parciales. |
Devoluciones
- una nueva instancia de BoostedTreesTrainingPredict
pública de salida <Integer> nodeIds ()
Tensor de rango 1 que contiene nuevos ID de nodo en los nuevos tree_ids.
pública Salida <> Float partialLogits ()
Rank 2 Tensor que contiene la actualización de logits (con respecto a los valores almacenados en caché) para cada ejemplo.
pública de salida <Integer> treeIds ()
Tensor de rango 1 que contiene nuevos identificadores de árbol para cada ejemplo.