一种基于梯度提升树算法的音乐点击率预测方法及装置

摘要

一种基于梯度提升树算法的音乐点击率预测方法及装置，属于机器学习技术领域，用以解决现有技术中音乐平台不能根据用户的个体特征和偏好变化进行高质量的音乐作品推荐的问题。本发明方法的技术要点包括，获取音乐数据训练集和测试集并分别提取音乐特征和用户特征；对提取的音乐特征和用户特征进行处理；对处理后的特征数据进行数据格式统一，获取训练特征数据集和待预测测试特征数据集；利用梯度提升树算法对训练特征数据集进行训练，获取预测模型；根据预测模型对待预测测试特征数据集进行预测分类。本发明可用于音乐平台针对不同的用户推荐不同的音乐作品。