Sklearn-使用交叉验证的价格预测

Question

这是我的代码：

from sklearn.datasets import load_boston
import matplotlib.pyplot as plt
import numpy as np
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import cross_val_predict
from sklearn.model_selection import train_test_split

%matplotlib inline

boston_properties = load_boston()

l_distance = boston_properties['data'][:, np.newaxis, 7]
linreg = LinearRegression()

X_train, X_test, y_train, y_test = train_test_split(l_distance, boston_properties['target'], test_size = 0.3)
y_pred = cross_val_predict(linreg, l_distance, boston_properties.target, cv=5)

plt.scatter(X_test, y_test, color='black')
plt.plot(X_test, y_pred, color='blue', linewidth=2)
plt.show()
print(y_pred.shape)

我收到的错误如下：

  

ValueError：x和y必须具有相同的第一尺寸，但形状为（152，1）和（506，）

我该如何做？

Answer 1

您创建了一个train_test_split，但您没有使用它来训练模型。然后，您可以预测整个训练数据，并将其与y_test进行比较。这是没有道理的。请改用这些行：

l_distance = boston_properties['data'][:, np.newaxis, 7]
linreg = LinearRegression()

X_train, X_test, y_train, y_test = train_test_split(l_distance,
    boston_properties['target'], test_size = 0.3) # now you have a train/test set
y_pred = cross_val_predict(linreg, X_train, y_train, cv=5) 

plt.scatter(X_train, y_train, color='black')
plt.plot(X_train, y_pred, color='blue', linewidth=2)
plt.show()

编辑：您还可以使用以下直线在各点之间形成一条直线：

plt.scatter(X_train, y_train, color='black')
plt.plot([X_train[np.argmin(X_train)], X_train[np.argmax(X_train)]],
         [y_pred[np.argmin(X_train)], y_pred[np.argmax(X_train)]],
         color='blue')
plt.show()