1. 程式人生 > >《機器學習實戰》第8章 ValueError: Masked arrays must be 1-D

《機器學習實戰》第8章 ValueError: Masked arrays must be 1-D

《機器學習實戰》第8章 預測數值型資料:迴歸 

函式regression1方法中出錯

def regression1():
    xArr, yArr = loadDataSet("./ex0.txt")
    xMat = mat(xArr)
    yMat = mat(yArr)
    ws = standRegres(xArr, yArr)
    fig = plt.figure()
    ax = fig.add_subplot(111)               
    ax.scatter(xMat[:, 1].flatten().tolist()

, yMat.T[:, 0].flatten().A[0].tolist()
    # 【原】ax.scatter(xMat[:, 1].flatten(), yMat.T[:, 0].flatten().A[0])

    xCopy = xMat.copy() 
    xCopy.sort(0)
    yHat = xCopy * ws
    ax.plot(xCopy[:, 1], yHat)
    plt.show()

修改後執行成功:

報錯