Beijing Baidu Netcom Science Technology Co., Ltd

State Key Laboratory of Internet of Things for Smart City (university of Macau)

Chengzhong Xu

Dejing Dou

Xingjian Li

Haoyi Xiong

Kafeng Wang

A method and an apparatus for training a classification model and data classification includes: obtaining a sample set and a pre-trained classification model, wherein the classification model includes at least two convolutional layers, each convolutional layer is connected to a classification layer through a fully connected layer; inputting the sample set into the classification model, and obtaining a prediction result output by each classification layer, wherein the prediction result includes a prediction probability of a class to which each sample belongs; calculating a probability threshold of each classification layer based on the prediction result output by each classification layer; setting a prediction stopping condition for the classification mode according to the probability threshold of each classification layer.

Growing community of inventors

S.A.R., China

Chengzhong Xu