Как нарисовать прогнозирующую плоскость на графике множественной линейной регрессии? ValueError: формы (2744,3) и (2,1) не выровнены: 3 (тусклый 1)! = 2 (тусклый 0) - PullRequest
Новая
       27

Как нарисовать прогнозирующую плоскость на графике множественной линейной регрессии? ValueError: формы (2744,3) и (2,1) не выровнены: 3 (тусклый 1)! = 2 (тусклый 0)

0 голосов
/ 23 апреля 2020

Я хочу нарисовать плоскость в Графике множественной линейной регрессии.

Мой фрейм данных состоит из опыта, возраста и зарплаты. Я хочу прогнозировать зарплату с точки зрения опыта и заработной платы.

Вот мой график данных, показанный ниже 

    experince   salary  age
0   0.5         2500    22
1   0.0         2250    21 
2   1.5         2800    25
.   .           .       .

Вот мой процесс изменения формы 

x = df.iloc[:,[0,2]].values
y = df.salary.values.reshape(-1,1)

У меня нет Идея, как нарисовать прогнозирующую плоскость в 3D-форме куба. Как я могу сделать этот процесс?

Я пытался написать этот код, но он не работает. Код основан на плоскости чертежа. 

a = np.linalg.solve(np.dot(x.T, x), np.dot(y.T, y))
predictedY = np.dot(x, a)
X = df.iloc[:,[0,1,2]].values
xx, yy, zz = np.meshgrid(X[:, 0], X[:, 1], X[:, 2]) <- ERROR HERE
combinedArrays = np.vstack((xx.flatten(), yy.flatten(), zz.flatten())).T
Z = combinedArrays.dot(a)

Ошибка показана ниже. 

ValueError: shapes (2744,3) and (2,1) not aligned: 3 (dim 1) != 2 (dim 0)
...