Classification network training method and device, computer equipment and storage medium

The invention relates to a classification network training method and a device, computer equipment and a storage medium. The method comprises the steps of determining a first weight factor of each category of sample data in a sample set according to the number of each category of sample data in the...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
1. Verfasser: AI JIANGBO
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:The invention relates to a classification network training method and a device, computer equipment and a storage medium. The method comprises the steps of determining a first weight factor of each category of sample data in a sample set according to the number of each category of sample data in the sample set; determining a second weight factor of each piece of sample data according to the gradient value of each piece of sample data in the sample set; and determining a loss value of the classification network according to the first weight factor, the second weight factor, the label corresponding to each piece of sample data and the classification result corresponding to each piece of sample data, and updating parameters in the classification network according to the loss value. According to the method, the value of the loss function containing the first weight factor and the second weight factor is adopted to train the classification network; and meanwhile, the problems of class imbalance of the training samp