Aracılığıyla paylaş


TextClassificationTrainer Sınıf

Tanım

Metni IEstimator<TTransformer> sınıflandırmak için Derin Sinir Ağı(DNN) eğitimi için.

public class TextClassificationTrainer : Microsoft.ML.TorchSharp.NasBert.NasBertTrainer<uint,long>
type TextClassificationTrainer = class
    inherit NasBertTrainer<uint32, int64>
Public Class TextClassificationTrainer
Inherits NasBertTrainer(Of UInteger, Long)
Devralma

Açıklamalar

Bu eğitmeni oluşturmak için TextClassification kullanın.

Giriş ve Çıkış Sütunları

Giriş etiketi sütun verileri anahtar türünde olmalı ve tümce sütunları türündeTextDataViewType olmalıdır.

Bu eğitmen aşağıdaki sütunları oluşturur:

Çıkış Sütunu Adı Sütun Türü Description
PredictedLabel anahtar türü Tahmin edilen etiketin dizini. Değeri i ise, gerçek etiket anahtar değerli giriş etiketi türündeki i. kategori olacaktır.
Score VektörSingle Tüm sınıfların puanları. Daha yüksek değer, ilişkili sınıfa düşme olasılığının yüksek olduğu anlamına gelir. i-th öğesi en büyük değere sahipse, tahmin edilen etiket dizini i.I'nin sıfır tabanlı dizin olduğuna dikkat edin.

Eğitmen Özellikleri

Makine öğrenmesi görevi Çok sınıflı sınıflandırma
Normalleştirme gerekli mi? No
Önbelleğe alma gerekli mi? No
Microsoft.ML ek olarak gerekli NuGet Microsoft.ML.TorchSharp ve libtorch-cpu veya libtorch-cuda-11.3 ya da işletim sistemine özgü varyantlardan herhangi biri.
ONNX'e aktarılabilir No

Eğitim Algoritması Ayrıntıları

Metni sınıflandırmak amacıyla önceden eğitilmiş mevcut bir NAS-BERT roBERTa modelinden yararlanarak derin sinir ağı (DNN) eğiter.

Yöntemler

Fit(IDataView)

Metni IEstimator<TTransformer> sınıflandırmak için Derin Sinir Ağı(DNN) eğitimi için.

(Devralındığı yer: TorchSharpBaseTrainer<TLabelCol,TTargetsCol>)
GetOutputSchema(SchemaShape)

Metni IEstimator<TTransformer> sınıflandırmak için Derin Sinir Ağı(DNN) eğitimi için.

(Devralındığı yer: NasBertTrainer<TLabelCol,TTargetsCol>)

Şunlara uygulanır