Personalized recommendation system based on matrix factorization and XGBoost algorithm
CSTR:
Author:
Affiliation:

Clc Number:

TP301.6

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In order to solve the data sparsity problem and improve the recommendation accuracy, we proposed a new matrix factorization XGBoost(MFXGB) recommendation algorithm which combined the matrix factorization method and the XGBoost (Extreme Gradient Boosting) algorithm. SVD++ algorithm (SVD, singular value decomposition) was used to fill the user-item score matrix to lessen the influence on the accuracy of the algorithm due to too many missing values and XGBoost was used to build a supervised learning model to predict the user's score. To reduce the computation time, feature extration based on the K-means clustering method was proposed to train XGBoot. The proposed MFXGB algorithm was applied to MovieLens dataset for experimental analysis and the results show that the recommendation accuracy was improved by 8.91%,10.18% and 11.79% respectively, compared with the three traditional algorithms.

    Reference
    Related
    Cited by
Get Citation

何婧,胡杰.融合矩阵分解和XGBoost的个性化推荐算法[J].重庆大学学报,2021,44(1):78~87

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:July 06,2020
  • Revised:
  • Adopted:
  • Online: January 08,2021
  • Published:
Article QR Code