在 Python 中擬合此線性回歸時出現問題

Question

我試圖使用兩個數組來練習我想繪制它們的位置 - 首先查看它們，然后創建一個線性回歸模型，然后擬合數據並預測它。

最后，我想繪制將它們分開的線性回歸線，但我不能 - 它沒有繪制它。 這是我的代碼：

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作為 x。

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