码迷,mamicode.com
首页 > 其他好文 > 详细

《机器学习实战》第7章的一处代码错误

时间:2018-03-19 22:43:18      阅读:616      评论:0      收藏:0      [点我收藏+]

标签:sla   python   learn   技术   ipython   执行   ipy   document   导致   

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-69-fab03f1fce4e> in <module>()
----> 1 adaboost.buildStump(dataArr, labelArr, D)

/home/cavin/Documents/BookData/machinelearninginaction/Ch07/adaboost.py in buildStump(dataArr, classLabels, D)
     44                 predictedVals = stumpClassify(dataMatrix, i, threshVal, inequal)
     45                 errArr = np.mat(np.ones((m, 1)))
---> 46                 errArr[predictedVals == labelMat] = 0
     47                 weightedError = D.T * errArr
     48                 print("""split: dim %d, thresh %.2f, thresh inequal: 

IndexError: index 1 is out of bounds for axis 1 with size 1

P120 程序清单7-1 单层决策树生成函数

def buildStump(dataArr, classLabels, D):
    ...
    errArr[predictedVals == labelMat] = 0

在3.6版本的anaconda下执行会提示错误:
技术分享图片

解决方法:

errArr[predictedVals == labelMat.T] = 0

也就是把labelMat转置,出现这个错误原因是predictedValslabelMat的shape不同导致的

《机器学习实战》第7章的一处代码错误

标签:sla   python   learn   技术   ipython   执行   ipy   document   导致   

原文地址:https://www.cnblogs.com/wuzhiblog/p/8605532.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!