Executa vários preditores de conjunto de regressão aditiva em instâncias de entrada e
calcula a atualização para logits armazenados em cache. Ele foi projetado para ser usado durante o treinamento. Ele percorre as árvores começando pelo ID da árvore em cache e pelo ID do nó em cache e calcula as atualizações a serem enviadas para o cache.
Métodos Públicos
BoostedTreesTrainingPredict estático | |
Saída <Inteiro> | nodeIds () Tensor de classificação 1 contendo novos ids de nó no novo tree_ids. |
Saída <flutuante> | parcialLogits () Tensor de classificação 2 contendo atualização de logits (em relação aos valores armazenados em cache) para cada exemplo. |
Saída <Inteiro> | treeIds () Tensor de classificação 1 contendo novos IDs de árvore para cada exemplo. |
Métodos herdados
Métodos Públicos
public static BoostedTreesTrainingPredict create ( Escopo de escopo , Operando <?> treeEnsembleHandle, Operando <Integer> cachedTreeIds, Operando <Integer> cachedNodeIds, Iterable< Operando <Integer>> bucketizedFeatures, Long logitsDimension)
Método de fábrica para criar uma classe que envolve uma nova operação BoostedTreesTrainingPredict.
Parâmetros
escopo | escopo atual |
---|---|
cachedTreeIds | Tensor de classificação 1 contendo IDs de árvores em cache, que é a árvore inicial de previsão. |
cachedNodeIds | Tensor de classificação 1 contendo o ID do nó em cache que é o nó inicial da previsão. |
recursos bucketizados | Uma lista de tensores de classificação 1 contendo ID de bucket para cada recurso. |
logitsDimensão | escalar, dimensão dos logits, a ser usada para forma de logits parciais. |
Devoluções
- uma nova instância de BoostedTreesTrainingPredict
Saída pública <Inteiro> nodeIds ()
Tensor de classificação 1 contendo novos ids de nó no novo tree_ids.
Saída pública <Float> parcialLogits ()
Tensor de classificação 2 contendo atualização de logits (em relação aos valores armazenados em cache) para cada exemplo.
Saída pública <Inteiro> treeIds ()
Tensor de classificação 1 contendo novos IDs de árvore para cada exemplo.