TextClassificationTrainer Třída
Definice
Důležité
Některé informace platí pro předběžně vydaný produkt, který se může zásadně změnit, než ho výrobce nebo autor vydá. Microsoft neposkytuje žádné záruky, výslovné ani předpokládané, týkající se zde uváděných informací.
Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) ke klasifikaci textu.
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)- Dědičnost
-
TextClassificationTrainer
Poznámky
K vytvoření tohoto školitele použijte TextClassification.
Vstupní a výstupní sloupce
Data sloupce vstupního popisku musí být typu klíč a sloupce vět musí být typuTextDataViewType .
Výstupem tohoto školitele jsou následující sloupce:
| Název výstupního sloupce | Typ sloupce | Description |
|---|---|---|
PredictedLabel |
typ klíče | Index predikovaného popisku Pokud je hodnota i, skutečný popisek by byl i-tý kategorie vstupního popisku s hodnotou klíče. |
Score |
Vektor zSingle | Skóre všech tříd. Vyšší hodnota znamená vyšší pravděpodobnost, že spadá do přidružené třídy. Pokud má element i-tý největší hodnotu, bude predikovaný index popisku i.Všimněte si, že i je index založený na nule. |
Charakteristiky trenéra
| Úloha strojového učení | Klasifikace s více třídami |
| Vyžaduje se normalizace? | No |
| Vyžaduje se ukládání do mezipaměti? | No |
| Kromě Microsoft.ML se vyžaduje NuGet. | Microsoft.ML.TorchSharp a libtorch-cpu nebo libtorch-cuda-11.3 nebo některou z variant specifických pro operační systém. |
| Exportovatelné do ONNX | No |
Podrobnosti trénovacího algoritmu
Trénuje hloubkovou neurální síť (DNN) využitím existujícího předem natrénovaného modelu nas-BERT roBERTa pro účely klasifikace textu.
Metody
| Fit(IDataView) |
Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) ke klasifikaci textu. (Zděděno od TorchSharpBaseTrainer<TLabelCol,TTargetsCol>) |
| GetOutputSchema(SchemaShape) |
Pro IEstimator<TTransformer> trénování hluboké neurální sítě (DNN) ke klasifikaci textu. (Zděděno od NasBertTrainer<TLabelCol,TTargetsCol>) |