Проблема при подгонке этой линейной регрессии в Python

Question

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

Наконец, я хотел нарисовать линию линейной регрессии, которая разделяет их, но я не в состоянии - она ​​этого не рисует. Это мой код:

k_true = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
k_pred = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,21]
plt.scatter(k_true,k_pred)
model = LinearRegression()
model.fit([k_true],[k_pred])
predictor = model.predict([k_pred])
plt.scatter(k_true,k_pred)
plt.plot([k_true],predictor,color="red")

Это результат

Answer 1

Вы строили неверную независимую переменную, а также predictor необходимо заменить на predictor[0], поскольку она содержала список в списке. Вам нужно использовать k_true как х.

plt.scatter(k_true, k_pred)
plt.plot(k_pred, predictor[0], '-k');