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 ser utilizado durante el entrenamiento. Atraviesa los árboles a partir de la identificación del árbol almacenado en caché y la identificación del nodo almacenado en caché y calcula las actualizaciones que se enviarán al caché.
Constantes
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
estático 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 envuelve una nueva operación BoostedTreesTrainingPredict. |
Salida < TInt32 > | ID de nodo () Tensor de rango 1 que contiene nuevos ID de nodo en el nuevo tree_ids. |
Salida <TFloat32> | Logits parciales () Tensor de rango 2 que contiene la actualización de logits (con respecto a los valores almacenados en caché) para cada ejemplo. |
Salida < TInt32 > | ID de árbol () Tensor de rango 1 que contiene nuevos identificadores de árbol para cada ejemplo. |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
creación pública estática de BoostedTreesTrainingPredict (alcance del 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 envuelve una nueva operación BoostedTreesTrainingPredict.
Parámetros
alcance | alcance actual |
---|---|
ID de árbol en caché | 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 almacenado en caché, que es el nodo inicial de la predicción. |
Características en cubos | 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 < TInt32 > nodeIds ()
Tensor de rango 1 que contiene nuevos ID de nodo en el nuevo tree_ids.
Salida pública <TFloat32> partLogits ()
Tensor de rango 2 que contiene la actualización de logits (con respecto a los valores almacenados en caché) para cada ejemplo.