NerTrainer Classe
Definição
Importante
Algumas informações se referem a produtos de pré-lançamento que podem ser substancialmente modificados antes do lançamento. A Microsoft não oferece garantias, expressas ou implícitas, das informações aqui fornecidas.
O IEstimator<TTransformer> para treinar uma DNN (Rede Neural Profunda) para classificar o texto.
public class NerTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<Microsoft.ML.Data.VBuffer<uint>,Microsoft.ML.Data.VBuffer<long>>type NerTrainer = class
inherit NasBertTrainer<VBuffer<uint32>, VBuffer<int64>>Public Class NerTrainer
Inherits NasBertTrainer(Of VBuffer(Of UInteger), VBuffer(Of Long))- Herança
Comentários
Para criar esse treinador, use NER.
Colunas de entrada e saída
Os dados da coluna de rótulo de entrada devem ser um Vetor do tipo de cadeia de caracteres e as colunas de frase devem ser do tipoTextDataViewType .
Este treinador gera as seguintes colunas:
| Nome da Coluna de Saída | Tipo de coluna | Descrição |
|---|---|---|
PredictedLabel |
Vetor do tipo de chave | O índice do rótulo previsto. Se seu valor for i, o rótulo real será a iº categoria no tipo de rótulo de entrada com valor de chave. |
| -- | -- | |
| Ferramenta de machine learning | Classificação multiclasse | |
| A normalização é necessária? | No | |
| O cache é necessário? | No | |
| NuGet necessário além de Microsoft.ML | Microsoft.ML.TorchSharp e libtorch-cpu ou libtorch-cuda-11.3 ou qualquer uma das variantes específicas do sistema operacional. | |
| Exportável para ONNX | No |
Detalhes do algoritmo de treinamento
Treina uma DNN (Rede Neural Profunda) aproveitando um modelo roBERTa NAS-BERT pré-treinado para fins de reconhecimento de entidade nomeada.
Métodos
| Fit(IDataView) |
O IEstimator<TTransformer> para treinar uma DNN (Rede Neural Profunda) para classificar o texto. (Herdado de TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
O IEstimator<TTransformer> para treinar uma DNN (Rede Neural Profunda) para classificar o texto. (Herdado de NasBertTrainer<TLabelCol,TTargetsCol>) |