我有一些关于sklearn linear_model.LinearRegression(模块)的输入问题。Sklearn线性回归X和Y输入格式
ages_train = [[20, 10000], [22, 12000], [22, 14000], [25, 17000], [30, 29000]]
net_worths_train = [10000, 12000, 14000, 17000, 29000]
reg = linear_model.LinearRegression()
reg.fit(ages_train, net_worths_train)
上述工作,但它使得很难绘图。举例来说,这样做是行不通的:
plt.scatter(ages_train, net_worths_train, color = "b", label = "train data")
所以我想做的是输入这样的:
ages_train = np.array([20, 22, 25, 27, 30, 31, 31, 34, 42, 50])
net_worths_train = np.array([10000, 12000, 16000, 20000, 30000, 33000, 34000, 38000, 49000, 60000])
但这种数据格式不与.fit工作() linear_model.LinearRegression的函数。输入数据的正确方法是什么,并使其易于绘图?最好是使用numpy数组吗?