Affiliations: Department of Electronic Information Engineering, Xi’an Technological University, Xi’an, China
Correspondence:
[*]
Corresponding author. Peng Wang, Department of Electronic Information Engineering, Xi’an Technological University, Xi’an 710021, China. E-mail: [email protected].
Abstract: In order to overcome the problems of poor accuracy and high complexity of current classification algorithm for non-equilibrium data set, this paper proposes a decision tree classification algorithm for non-equilibrium data set based on random forest. Wavelet packet decomposition is used to denoise non-equilibrium data, and SNM algorithm and RFID are combined to remove redundant data from data sets. Based on the results of data processing, the non-equilibrium data sets are classified by random forest method. According to Bootstrap resampling method with certain constraints, the majority and minority samples of each sample subset are sampled, CART is used to train the data set, and a decision tree is constructed. Obtain the final classification results by voting on the CART decision tree classification. Experimental results show that the proposed algorithm has the characteristics of high classification accuracy and low complexity, and it is a feasible classification algorithm for non-equilibrium data set.
Keywords: Random forest, non-equilibrium data set, decision tree, classification, SNM algorithm, RFID
