FastTreeBinaryTrainer Classe

Definição

Para IEstimator<TTransformer> treinar um modelo de classificação binária de árvore de decisão usando o FastTree.

public sealed class FastTreeBinaryTrainer : Microsoft.ML.Trainers.FastTree.BoostingFastTreeTrainerBase<Microsoft.ML.Trainers.FastTree.FastTreeBinaryTrainer.Options,Microsoft.ML.Data.BinaryPredictionTransformer<Microsoft.ML.Calibrators.CalibratedModelParametersBase<Microsoft.ML.Trainers.FastTree.FastTreeBinaryModelParameters,Microsoft.ML.Calibrators.PlattCalibrator>>,Microsoft.ML.Calibrators.CalibratedModelParametersBase<Microsoft.ML.Trainers.FastTree.FastTreeBinaryModelParameters,Microsoft.ML.Calibrators.PlattCalibrator>>
type FastTreeBinaryTrainer = class
    inherit BoostingFastTreeTrainerBase<FastTreeBinaryTrainer.Options, BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>, CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>
Public NotInheritable Class FastTreeBinaryTrainer
Inherits BoostingFastTreeTrainerBase(Of FastTreeBinaryTrainer.Options, BinaryPredictionTransformer(Of CalibratedModelParametersBase(Of FastTreeBinaryModelParameters, PlattCalibrator)), CalibratedModelParametersBase(Of FastTreeBinaryModelParameters, PlattCalibrator))
Herança

Comentários

Para criar esse treinador, use FastTree ou FastTree(Options).

Colunas de entrada e saída

Os dados da coluna de rótulo de entrada devem ser Boolean. Os dados da coluna de recursos de entrada devem ser um vetor de tamanho conhecido de Single.

Este treinador gera as seguintes colunas:

Nome da coluna de saída Tipo de Coluna Descrição
Score Single A pontuação não associado calculada pelo modelo.
PredictedLabel Boolean O rótulo previsto com base no sinal da pontuação. Uma pontuação negativa é mapeada para false e uma pontuação positiva é mapeada para true.
Probability Single A probabilidade calculada calibrando a pontuação de ter true como o rótulo. O valor da probabilidade está no intervalo [0, 1].

Características do Treinador

Tarefa de machine learning Classificação binária
A normalização é necessária? No
O cache é necessário? No
NuGet necessário além de Microsoft.ML Microsoft.ML.FastTree
Exportável para ONNX Yes

Detalhes do algoritmo de treinamento

FastTree é uma implementação eficiente do algoritmo de aumento de gradiente MART . O aumento de gradiente é uma técnica de aprendizado de máquina para problemas de regressão. Ele cria cada árvore de regressão de forma passo a passo, usando uma função de perda predefinida para medir o erro para cada etapa e corrige para ela na próxima. Portanto, esse modelo de previsão é, na verdade, um conjunto de modelos de previsão mais fracos. Em problemas de regressão, o aumento cria uma série dessas árvores de forma passo a passo e seleciona a árvore ideal usando uma função de perda arbitrariamente diferencial.

O MART aprende um conjunto de árvores de regressão, que é uma árvore de decisão com valores escalares em suas folhas. Uma árvore de decisão (ou regressão) é um fluxograma binário semelhante a uma árvore, em que em cada nó interior um decide qual dos dois nós filho continuará com base em um dos valores de recurso da entrada. Em cada nó folha, um valor é retornado. Nos nós interiores, a decisão é baseada no teste x <= v em que x é o valor do recurso no exemplo de entrada e v é um dos valores possíveis desse recurso. As funções que podem ser produzidas por uma árvore de regressão são todas as funções constantes em termos de peça.

O conjunto de árvores é produzido pela computação, em cada etapa, de uma árvore de regressão que aproxima o gradiente da função de perda e adicionando-a à árvore anterior com coeficientes que minimizam a perda da nova árvore. A saída do conjunto produzido pelo MART em uma determinada instância é a soma das saídas de árvore.

  • No caso de um problema de classificação binária, a saída é convertida em uma probabilidade usando alguma forma de calibragem.
  • No caso de um problema de regressão, a saída é o valor previsto da função.
  • No caso de um problema de classificação, as instâncias são ordenadas pelo valor de saída do conjunto.

Para obter mais informações, consulte:

Verifique a seção Consulte Também para obter links para exemplos do uso.

Campos

Nome Description
FeatureColumn

A coluna de recursos que o treinador espera.

(Herdado de TrainerEstimatorBase<TTransformer,TModel>)
GroupIdColumn

A coluna groupID opcional esperada pelos treinadores de classificação.

(Herdado de TrainerEstimatorBaseWithGroupId<TTransformer,TModel>)
LabelColumn

A coluna de rótulo que o treinador espera. Pode ser null, o que indica que o rótulo não é usado para treinamento.

(Herdado de TrainerEstimatorBase<TTransformer,TModel>)
WeightColumn

A coluna de peso que o treinador espera. Pode ser null, o que indica que o peso não é usado para treinamento.

(Herdado de TrainerEstimatorBase<TTransformer,TModel>)

Propriedades

Nome Description
Info

Para IEstimator<TTransformer> treinar um modelo de classificação binária de árvore de decisão usando o FastTree.

(Herdado de FastTreeTrainerBase<TOptions,TTransformer,TModel>)

Métodos

Nome Description
Fit(IDataView, IDataView)

Treina um FastTreeBinaryTrainer usando dados de treinamento e validação, retorna um BinaryPredictionTransformer<TModel>.

Fit(IDataView)

Treina e retorna um ITransformer.

(Herdado de TrainerEstimatorBase<TTransformer,TModel>)
GetOutputSchema(SchemaShape)

Para IEstimator<TTransformer> treinar um modelo de classificação binária de árvore de decisão usando o FastTree.

(Herdado de TrainerEstimatorBase<TTransformer,TModel>)

Métodos de Extensão

Nome Description
AppendCacheCheckpoint<TTrans>(IEstimator<TTrans>, IHostEnvironment)

Acrescente um 'ponto de verificação de cache' à cadeia do avaliador. Isso garantirá que os avaliadores downstream serão treinados em relação aos dados armazenados em cache. É útil ter um ponto de verificação de cache antes dos treinadores que têm várias passagens de dados.

WithOnFitDelegate<TTransformer>(IEstimator<TTransformer>, Action<TTransformer>)

Dado um avaliador, retorne um objeto de encapsulamento que chamará um delegado uma vez Fit(IDataView) chamado. Geralmente, é importante que um avaliador retorne informações sobre o que estava em forma, e é por isso que o Fit(IDataView) método retorna um objeto especificamente tipado, em vez de apenas um geral ITransformer. No entanto, ao mesmo tempo, IEstimator<TTransformer> muitas vezes são formados em pipelines com muitos objetos, portanto, talvez seja necessário criar uma cadeia de avaliadores por meio EstimatorChain<TLastTransformer> de onde o avaliador para o qual queremos obter o transformador está enterrado em algum lugar nessa cadeia. Para esse cenário, podemos por meio desse método anexar um delegado que será chamado assim que o ajuste for chamado.

Aplica-se a

Confira também