首页    期刊浏览 2025年02月22日 星期六
登录注册

文章基本信息

  • 标题:Evaluation of Company Investment Value based on Machine Learning
  • 本地全文:下载
  • 作者:Junfeng Hu ; Xiaosa Li ; Yuru Xu
  • 期刊名称:Computer Science & Information Technology
  • 电子版ISSN:2231-5403
  • 出版年度:2020
  • 卷号:10
  • 期号:11
  • 页码:75-81
  • DOI:10.5121/csit.2020.101107
  • 出版社:Academy & Industry Research Collaboration Center (AIRCC)
  • 摘要:In this paper, company investment value evaluation models are established based on comprehensive company information. After data mining and extracting a set of 436 feature parameters, an optimal subset of features is obtained by dimension reduction through treebased feature selection, followed by the 5-fold cross-validation using XGBoost and LightGBM models. The results show that the Root-Mean-Square Error (RMSE) reached 3.098 and 3.059, respectively. In order to further improve the stability and generalization capability, Bayesian Ridge Regression has been used to train a stacking model based on the XGBoost and LightGBM models. The corresponding RMSE is up to 3.047. Finally, the importance of different features to the LightGBM model is analysed.
  • 关键词:Company investment value assessment ;XGBoost model ;LightGBM model ;Model fusion.
国家哲学社会科学文献中心版权所有