推荐系统系列3-基于决策树做推荐系统的方法

标签：代码   推荐系统   文本分类   概念   比赛   enter   学习方法   数据挖掘   end   

    

3、xgboost

    XGBoost（eXtreme Gradient Boosting）全名叫极端梯度提升，本质上是一种高级的决策树（其实应该较回归树）。XGBoost是集成学习方法的王牌，在Kaggle数据挖掘比赛中，大部分获胜者用了XGBoost，XGBoost在绝大多数的回归和分类问题上表现的十分顶尖。

 

4、

    参考资料：《西瓜书》

    参考代码：https://github.com/dongguadan/recommender-system/tree/master/xgboost-Based_Filter

    数据集采用bbc提供的文本语料库：bbc.terms、bbc.classes、bbc.mtx。代码根据语料库将数据组织成特征矩阵，如图3：

                                                                                                                               图3

    然后将特征矩阵随机划分成训练集（70%）、测试集（30%），进行训练、测试：文章共分为5类分别用0-4来表示；图4表示每篇文章所属分类；图5表示每篇文章归入某一类别的概率。

                                                                                                                                  图4

                                                                                                                                  图5

推荐系统系列3-基于决策树做推荐系统的方法

标签：代码   推荐系统   文本分类   概念   比赛   enter   学习方法   数据挖掘   end   

原文地址：https://www.cnblogs.com/dongguadan/p/10981602.html