Journal of Beijing University of Posts and Telecommunications

  • EI核心期刊

Journal of Beijing University of Posts and Telecommunications ›› 2021, Vol. 44 ›› Issue (5): 81-87,106.doi: 10.13190/j.jbupt.2021-010

• PAPERS • Previous Articles     Next Articles

Multi-Modal Transportation Recommendation Based on Graph Embedding and CaGBDT

SUN Quan-ming, CHANG Lei, MA Cheng, QU Zhi-jian   

  1. School of Computer Science and Technology, Shandong University of Technology, Zibo 255049, China
  • Received:2021-01-19 Online:2021-10-28 Published:2021-09-06

Abstract: To solve the problems in transportation recommendation service,such as single recommended methods and ignoring the user travel preferences,a cascade gradient boosting decision tree (CaGBDT) model is proposed based on the multi-grained cascade forest structure. CaGBDT uses the cascade structure to increase the depth, and then realizes the deep-level representation learning of features. Meanwhile to solve the imbalance of sample class,an index optimization layer based on Powell algorithm is proposed. By searching a threshold for each class,the weight of the prediction results of the model is modified to maximize the evaluation indexes. In addition,a user travel global relationship graph can be constructed by the CaGBDT model via referring the user's travel record, and the spatial context relationship of the user's travel is extracted automatically by the graph embedding method to improve the efficiency of feature extraction.

Key words: traffic recommendation, graph embedding, feature engineering, cascade forest, gradient boosting decision tree

CLC Number: