BoostedTreesTrainingPredict

BoostedTreesTrainingPredict public final class

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é.

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

estática BoostedTreesTrainingPredict
crear ( Alcance alcance, operando <?> treeEnsembleHandle, operando < TInt32 > cachedTreeIds, operando < TInt32 > cachedNodeIds, Iterable < operando < TInt32 >> bucketizedFeatures, Long logitsDimension)
Método de fábrica para crear una clase que envuelva una nueva operación BoostedTreesTrainingPredict.
Salida < TInt32 >
nodeIds ()
Tensor de rango 1 que contiene nuevos ID de nodo en los nuevos tree_ids.
Salida < TFloat32 >
partialLogits ()
Rank 2 Tensor que contiene la actualización de logits (con respecto a los valores almacenados en caché) para cada ejemplo.
Salida < TInt32 >
treeIds ()
Tensor de rango 1 que contiene nuevos identificadores de árbol para cada ejemplo.

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "BoostedTreesTrainingPredict"

Métodos públicos

public static BoostedTreesTrainingPredict crear ( Alcance alcance, operando <?> treeEnsembleHandle, operando < TInt32 > cachedTreeIds, operando < TInt32 > cachedNodeIds, Iterable < operando < TInt32 >> 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 < TInt32 > nodeIds ()

Tensor de rango 1 que contiene nuevos ID de nodo en los nuevos tree_ids.

pública de salida < TFloat32 > 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 < TInt32 > treeIds ()

Tensor de rango 1 que contiene nuevos identificadores de árbol para cada ejemplo.