TextClassificationTrainer Sınıf

Tanım

Metni IEstimator<TTransformer> sınıflandırmak için Bir Derin Sinir Ağı (DNN) eğitmek 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
Devralma
TextClassificationTrainer

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ünde TextDataViewTypeolmalıdır.

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

Çıkış sütun adı Sütun türü Description
PredictedLabel anahtar türü Tahmin edilen etiketin dizini. Değeri iise, gerçek etiket anahtar değerli giriş etiketi türündeki i-th kategorisi 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 iolacaktır. Sıfır tabanlı bir dizin olduğunu i unutmayın.

Eğitmen özellikleri

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

Eğitim algoritması ayrıntıları

Metni sınıflandırmak amacıyla mevcut, önceden eğitilmiş NAS-BERT roBERTa modelinden yararlanarak Derin Sinir Ağı (DNN) eğitilir.

Yöntemler

Name Description
Fit(IDataView)

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

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

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

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

Uzantı Metotları

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

Tahmin zincirine bir 'önbelleğe alma denetim noktası' ekler. Bu, aşağı akış tahmincilerinin önbelleğe alınmış verilere karşı eğitilmesini sağlar. Birden fazla veri alan eğitmenler geçişten önce bir önbelleğe alma kontrol noktasının olması yararlıdır.

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

Tahmin aracı verildiğinde, bir temsilci çağrıldıktan sonra Fit(IDataView) çağıracak bir sarmalama nesnesi döndürün. Tahmin aracının neyin uygun olduğu hakkında bilgi döndürmesi genellikle önemlidir. Bu nedenle Fit(IDataView) yöntem yalnızca genel ITransformerbir nesne yerine özel olarak yazılan bir nesne döndürür. Bununla birlikte, aynı zamanda, IEstimator<TTransformer> genellikle birçok nesne içeren işlem hatları halinde oluşturulur, bu nedenle transformatörü almak istediğimiz tahmin aracının EstimatorChain<TLastTransformer> bu zincirde bir yere gömüldüğü bir tahmin zinciri oluşturmamız gerekebilir. Bu senaryo için, bu yöntem aracılığıyla uygun çağrıldıktan sonra çağrılacak bir temsilci ekleyebiliriz.

Şunlara uygulanır