wzy6642/Machine-Learning-in-Action-Python3

BUG: Machine-Learning-in-Action-Python3/CART_Project3/CART.py

xiaoliou008 opened this issue · 1 comments

xiaoliou008 commented

Machine-Learning-in-Action-Python3/CART_Project3/CART.py
第285行
errorNoMerge = np.sum(np.power(lSet[:, -1] - tree['left'], 2)) + np.sum(np.power(rSet[:, 1] - tree['right'], 2))
中的后半部分rSet的索引似乎应该是[:, -1]。应修改为：
errorNoMerge = np.sum(np.power(lSet[:, -1] - tree['left'], 2)) + np.sum(np.power(rSet[:, -1] - tree['right'], 2))

wzy6642 commented

是这样子，谢谢指正