编号不一致的数组

时间:2019-06-04 23:47:30

标签: machine-learning python-3.6 linear-regression

这只是我正在复制的教程中的代码示例,并且不断出现此错误。有人可以给我一个明确而详细的解决方案。

import matplotlib.pyplot as plt
import numpy as np



from sklearn import datasets,linear_model


house_price=[245, 312, 279, 308, 199, 405, 324, 319, 255]
size= [1400, 1600, 1700, 1875, 1100, 1550, 2350, 2450, 1425, 1700]

size2 = np.array(size).reshape((-1,1))
print(size2)



regr = linear_model.LinearRegression()
regr.fit(size2, house_price)
print("coefficient: \n" ,regr.coef_)
print("intercept: \n", regr.intercept_)


def graph(formula, x_range):
    x = np.array(x_range)
    y = eval(formula)
    plt.plot(x,y)

1 个答案:

答案 0 :(得分:1)

房价数量与尺寸数量(数组中的元素)不匹配。他们需要匹配。否则,在缺少值的地方添加np.nan(或一些估算值,例如中位数)。例如:

house_price=[245, 312, 279, 308, 199, 405, 324, 319, 255, np.nan]
size= [1400, 1600, 1700, 1875, 1100, 1550, 2350, 2450, 1425, 1700]