所以我正在研究线性回归。到目前为止,我已经设法在线性回归中绘图,但目前我正在使用多元线性回归,我无法绘制它,如果我手动输入值,我可以获得一些结果,但我无法设法绘制它。下面是我的代码块和数据集和错误,我可以改变什么来绘制它?
数据集:
deneyim maas yas
0.5 2500 22
0 2250 21
1 2750 23
5 8000 25
8 9000 28
4 6900 23
15 20000 35
7 8500 29
3 6000 22
2 3500 23
12 15000 32
10 13000 30
14 18000 34
6 7500 27
代码块:
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.linear_model import LinearRegression
dataset = pd.read_csv("multiple-linear-regression-dataset.csv",sep = ";")
x = dataset.iloc[:,[0,2]].values
y = dataset.maas.values.reshape(-1,1)
multiple_lr = LinearRegression()
multiple_lr.fit(x,y)
b0 = multiple_lr.intercept_
b1 = multiple_lr.coef_
b2 = b1
multiple_lr.predict(np.array([[10,35],[5,35]]))
array = np.array([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]).reshape(-1,1)
y_head = multiple_lr.predict(array)
plt.scatter(x,y)
plt.plot(array, y_head, color = "red")
plt.show()
它说ValueError: shapes (16,1) and (2,1) not aligned: 1 (dim 1) != 2 (dim 0)当我尝试编译它时。