基于神经网络和树模型的预测模型建立方法及其装置

摘要

本发明公开了一种基于神经网络和树模型的预测模型建立方法及其装置，该方法包括获取数据集，对所述数据集进行预处理得到预处理数据集，并将所述预处理数据集分为训练数据集和测试数据集；根据预设神经网络模型和预设树模型构建神经网络树模型，根据所述神经网络树模型构建神经网络随机森林模型；将所述训练数据集输入至所述神经网络随机森林模型进行模型训练；将所述测试数据集输入至训练结束后的神经网络随机森林模型进行模型验证。本发明提供的基于神经网络和树模型的预测模型建立方法，将神经网络与树模型相结合得到神经网络随机森林模型，通过神经网络随机森林模型进行相关预测，不仅提高了预测的准确度，也提高了预测的效率。