Method, device, electronic device and storage medium for data classification model training

Abstract

The invention relates to the field of deep learning, in particular to a data classification model training method and device, an electronic device and a storage medium. The method comprises the following steps of acquiring a plurality of first sample data, a plurality of second sample data, a first iteration frequency and a second iteration frequency; within the first iteration frequency, on the basis of a plurality of first sample data, training the first data classification model in an annular training mode to obtain a second data classification model; within a second iteration frequency, based on a plurality of second sample data, using a tree training mode to train the second data classification model to obtain a third data classification model. At the data classification training, theannular training mode and the tree training mode are combined, so that a large amount of time can be saved compared with the mode of independently using the tree training mode, meanwhile, the accuracy of the data classification model can be guaranteed, and the training efficiency of the data classification model is improved.

Description

technical field [0001] The present disclosure relates to the field of deep learning, and in particular to a method, device, electronic device and storage medium for data classification model training. Background technique [0002] With the development of deep learning technology, people can effectively deal with machine classification problems such as speech recognition or image classification by using deep learning methods. When performing data classification, it is necessary to train the data classification model first, and perform data classification based on the data classification model. In order to improve the classification accuracy of the data classification model, it is often necessary to provide a large amount of sample data, and more sample data also means a heavy computational burden. In the field of deep learning training, compared with CPU (Central Processing Unit, central processing unit), GPU (Graphics Processing Unit, graphics processing unit) has great adv...

AI Technical Summary

Problems solved by technology

[0005] The present disclosure provides a method, device, electronic device and storage medium for training a data classification model, which can overcome the problem that the training data classification model takes up a lot of time, resulting in low efficiency of data classification model training

Images