Aviso: esta API está obsoleta e será removida em uma versão futura do TensorFlow depois que a substituição estiver estável.

BoostedTreesCalculateBestFeatureSplit

BoostedTreesCalculateBestFeatureSplit classe final pública

Calcula os ganhos para cada recurso e retorna as melhores informações de divisão possíveis para o recurso.

As informações de divisão são o melhor limite (id do intervalo), ganhos e contribuições de nó esquerdo / direito por nó para cada recurso.

É possível que nem todos os nós possam ser divididos em cada recurso. Portanto, a lista de nós possíveis pode diferir entre os recursos. Portanto, retornamos `node_ids_list` para cada recurso, contendo a lista de nós que este recurso pode ser usado para dividir.

Dessa forma, a saída é a melhor divisão por recursos e por nó, de modo que precisa ser combinada posteriormente para produzir a melhor divisão para cada nó (entre todos os recursos possíveis).

As formas de saída são compatíveis de forma que a primeira dimensão de todos os tensores seja a mesma e igual ao número de nós de divisão possíveis para cada recurso.

Classes aninhadas

classe BoostedTreesCalculateBestFeatureSplit.Options Atributos opcionais para BoostedTreesCalculateBestFeatureSplit

Métodos Públicos

estática BoostedTreesCalculateBestFeatureSplit
criar ( Scope escopo, Operando <Integer> nodeIdRange, Operando <float> statsSummary, Operando <float> L1, Operando <float> l2, Operando <float> treeComplexity, Operando <float> minNodeWeight, Long logitsDimension, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação BoostedTreesCalculateBestFeatureSplit.
Output <Integer>
featureDimensions ()
Um tensores Rank 1 indicando a melhor dimensão de recurso para cada recurso a ser dividido para determinados nós se o recurso for multidimensional.
Output <float>
ganhos ()
Tensores de Rank 1 indicando os melhores ganhos para cada recurso a ser dividido para determinados nós.
Output <float>
leftNodeContribs ()
Um tensor Rank 2 indicando a contribuição dos nós esquerdos ao ramificar de nós pais (dado pelo elemento tensor na saída node_ids_list) para a direção esquerda pelo limite fornecido para cada recurso.
Output <Integer>
nodeIds ()
Tensores de Rank 1 indicando possíveis ids de nós divididos para cada recurso.
Output <float>
rightNodeContribs ()
Tensores de Rank 2, com a mesma forma / condições que left_node_contribs_list, mas apenas que o valor é para o nó direito.
estáticos BoostedTreesCalculateBestFeatureSplit.Options
splitType (String splitType)
Output <String>
splitWithDefaultDirections ()
Tensores de Rank 1 indicando a direção a seguir se houver dados ausentes.
Output <Integer>
limiares ()
Tensores de Rank 1 indicando a id do intervalo para comparação (como um limite) para divisão em cada nó.

Métodos herdados

Métodos Públicos

public static BoostedTreesCalculateBestFeatureSplit criar ( Scope escopo, Operando <inteiro> nodeIdRange, Operando <float> statsSummary, Operando <float> L1, Operando <float> L2, Operando <float> treeComplexity, Operando <float> minNodeWeight, Longo logitsDimension, Opções .. . opções)

Método de fábrica para criar uma classe que envolve uma nova operação BoostedTreesCalculateBestFeatureSplit.

Parâmetros
alcance escopo atual
nodeIdRange Um tensor Rank 1 (forma = [2]) para especificar o intervalo [primeiro, último) de ids de nó para processar dentro de `stats_summary_list`. Os nós são iterados entre os dois nós especificados pelo tensor, como `para node_id no intervalo (node_id_range [0], node_id_range [1])` (Observe que o último índice node_id_range [1] é exclusivo).
statsSummary Um tensor Rank 4 (# shape = [max_splits, feature_dims, bucket, stats_dims]) para resumo de estatísticas acumuladas (gradiente / hessian) por nó, por dimensão, por intervalos para cada recurso. A primeira dimensão do tensor é o número máximo de divisões e, portanto, nem todos os elementos dele serão usados, mas apenas os índices especificados por node_ids serão usados.
l1 l1 fator de regularização nos pesos das folhas, por instância com base.
12 l2 fator de regularização nos pesos das folhas, por instância com base.
treeComplexity ajuste ao ganho, por folha.
minNodeWeight média mínima de hessianos em um nó antes do necessário para que o nó seja considerado para divisão.
logitsDimension A dimensão de logit, ou seja, número de classes.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de BoostedTreesCalculateBestFeatureSplit

pública Output <Integer> featureDimensions ()

Um tensores Rank 1 indicando a melhor dimensão de recurso para cada recurso a ser dividido para determinados nós se o recurso for multidimensional. Veja acima para detalhes como formas e tamanhos.

pública Output <float> ganhos ()

Tensores de Rank 1 indicando os melhores ganhos para cada recurso a ser dividido para determinados nós. Veja acima para detalhes como formas e tamanhos.

pública Output <float> leftNodeContribs ()

Um tensor Rank 2 indicando a contribuição dos nós esquerdos ao ramificar dos nós pais (dado pelo elemento tensor na saída node_ids_list) para a direção esquerda pelo limite fornecido para cada recurso. Este valor será usado para fazer o valor do nó esquerdo adicionando ao valor do nó pai. O tamanho da segunda dimensão é 1 para logits unidimensionais, mas seria maior para problemas de várias classes. Veja acima para detalhes como formas e tamanhos.

pública Output <Integer> nodeIds ()

Tensores de Rank 1 indicando possíveis ids de nós divididos para cada recurso. O comprimento da lista é num_features, mas cada tensor tem um tamanho diferente, pois cada recurso fornece diferentes nós possíveis. Veja acima para detalhes como formas e tamanhos.

pública Output <float> rightNodeContribs ()

Tensores de Rank 2, com a mesma forma / condições que left_node_contribs_list, mas apenas que o valor é para o nó direito.

public static BoostedTreesCalculateBestFeatureSplit.Options splitType (String splitType)

Parâmetros
splitType Uma string que indica se este Op deve realizar a divisão de desigualdade ou divisão de igualdade.

pública Output <String> splitWithDefaultDirections ()

Tensores do Rank 1 indicando a direção a seguir se houver dados ausentes. Veja acima para detalhes como formas e tamanhos. A desigualdade com o padrão à esquerda retorna 0, a desigualdade com o padrão à direita retorna 1, a igualdade com o padrão à direita retorna 2.

pública Output <Integer> limiares ()

Um tensores Rank 1 indicando a id do intervalo para comparação (como um limite) para divisão em cada nó. Veja acima para detalhes como formas e tamanhos.