The invention relates to the field of
machine learning, and especially relates to a multi-
label classification method based on a gravity model, comprising the steps of: acquiring a sample set with labels to serve as a training sample set; calculating the distances from one training sample to other training samples and performing sorting to obtain a
near neighbor set of the training sample; in thenear neighbor set, building a
positive correlation matrix based on
positive correlation properties among the labels, and building a
negative correlation matrix based on
negative correlation propertiesamong the labels; calculating a
near neighbor set of a to-be-detected sample, and building a to-be-detected
positive correlation matrix and a to-be-detected
negative correlation matrix according to the
near neighbor set; obtaining a positive correlation data granule and a negative correlation data granule based on the to-be-detected positive correlation matrix and the to-be-detected negative correlation matrix; and building the gravity model, and performing classification through a gravity relationship between the to-be-detected sample and the positive correlation data granule as well as thenegative correlation data granule. The multi-
label classification method of the invention has the beneficial effects that: consideration about a negative correlation relationship between the labels isintroduced, the correlation between the labels is taken full
advantage of, and the correlation relationship is mined in the near neighbor set, thereby avoiding global computing and reducing complexity.