中文

基于机器学习的公司投资价值评估

统计金融 2020-10-06 v1 机器学习

摘要

本文基于公司综合信息建立公司投资价值评估模型。在对数据进行挖掘并提取一组 436 个特征参数后,通过基于树的特征选择进行降维得到最优特征子集,随后使用 XGBoost 与 LightGBM 模型进行 5 折交叉验证。结果表明,均方根误差(RMSE)分别达到 3.098 与 3.059。为了进一步提高稳定性与泛化能力,采用贝叶斯岭回归基于 XGBoost 与 LightGBM 模型训练了一个堆叠模型。相应的 RMSE 达到 3.047。最后,分析了不同特征对 LightGBM 模型的重要性。

关键词

引用

@article{arxiv.2010.01996,
  title  = {Evaluation of company investment value based on machine learning},
  author = {Junfeng Hu and Xiaosa Li and Yuru Xu and Shaowu Wu and Bin Zheng},
  journal= {arXiv preprint arXiv:2010.01996},
  year   = {2020}
}