Lightweight classification model training method and device, medium and equipment
The invention provides a lightweight classification model training method and device, a medium and equipment. The method comprises the following steps: collecting first text data with a classification label and second text data without a classification label; according to the first text data, traini...
Gespeichert in:
Hauptverfasser: | , , , |
---|---|
Format: | Patent |
Sprache: | chi ; eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | The invention provides a lightweight classification model training method and device, a medium and equipment. The method comprises the following steps: collecting first text data with a classification label and second text data without a classification label; according to the first text data, training a large-scale language model to obtain a training convergent large-scale language model; inputting the second text data into the large-scale language model to obtain a classification label corresponding to the second text data, and labeling the second text data by using the classification label; and training a lightweight classification model on the edge device according to the labeled second text data. According to the embodiment of the invention, the problem that more accurate text classification is difficult to realize due to less marked data in a small scene is solved based on the existing large-scale language model. And meanwhile, the lightweight classification model has the advantage of real-time response. |
---|