BoostedTreesTrainingPredict

aula final pública BoostedTreesTrainingPredict

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
create ( Escopo do 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.
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.