Share via


NerTrainer Class

Definition

The IEstimator<TTransformer> for training a Deep Neural Network(DNN) to classify text.

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))
Inheritance

Remarks

To create this trainer, use NER.

Input and Output Columns

The input label column data must be a Vector of string type and the sentence columns must be of typeTextDataViewType.

This trainer outputs the following columns:

Output Column Name Column Type Description
PredictedLabel Vector of key type The predicted label's index. If its value is i, the actual label would be the i-th category in the key-valued input label type.
-- --
Machine learning task Multiclass classification
Is normalization required? No
Is caching required? No
Required NuGet in addition to Microsoft.ML Microsoft.ML.TorchSharp and libtorch-cpu or libtorch-cuda-11.3 or any of the OS specific variants.
Exportable to ONNX No

Training Algorithm Details

Trains a Deep Neural Network(DNN) by leveraging an existing pre-trained NAS-BERT roBERTa model for the purpose of named entity recognition.

Methods

Fit(IDataView) (Inherited from TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape) (Inherited from NasBertTrainer<TLabelCol,TTargetsCol>)

Applies to