Based on the breast cancer diagnosis data set of the kaggle website, machine learning algorithms are used to construct prediction models, the basic principles and model building methods are expounded, and the experimental results are analyzed. The classification accuracy of the trained prediction model is improved after dimensionality reduction, and the classification effect of the prediction model built by XGBoost is better.