您的位置:首页 > 编程语言

机器学习实战 第九章 的树回归这里代码有问题

2016-11-19 00:58 260 查看
我的文章来自@fghdvbgt的一篇博客:http://m.blog.csdn.net/article/details?id=51277716

发现里面确实有着两个问题:

(1):def binSplitDataSet(dataSet, feature, value):
mat0 = dataSet[nonzero(dataSet[:,feature] > value)[0],:][0]
mat1 = dataSet[nonzero(dataSet[:,feature] <= value)[0],:][0]
return mat0,mat1
mat0,mat1应去掉后面的[0],这样才能将原dataSet按照feature的种类分割
(2): 是chooseBestSplit函数中的: for splitVal in set(dataSet[:,featIndex]):应该改为:
for splitVal in set(i[0] for i in dataSet[:,index].tolist()):这样才可以正确运行


                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: