我正在尝试使用线性回归技术查找linnerud数据集的性能和均方误差。我在传递数据时陷入困境,并收到错误“ ValueError:找到的输入变量样本数量不一致:[10,1]”。 Linnerud数据集具有三个功能和目标中的三列,我只想使用chinup的一个功能。有人可以帮我解决我遇到的困难吗?
以下是我到目前为止所做的尝试,通过引用https://scikit-learn.org/stable/auto_examples/linear_model/plot_ols.html
from sklearn import datasets
from sklearn import linear_model
from sklearn.metrics import mean_squared_error,r2_score
import matplotlib.pyplot as plt
import numpy as np
linnerud = datasets.load_linnerud()
print(linnerud)
# Use only one feature
linnerud_X = linnerud.data[:,np.newaxis,0]
print(linnerud_X)
X = np.array(linnerud_X).reshape((1,-1))
print(X)
# Split the data into training/testing sets
linnerud_X_train = linnerud_X[:-10]
linnerud_X_test = linnerud_X[-10:]
#print(linnerud_X_train)
#print(linnerud_X_test)
Y = np.array(linnerud.target).reshape((1,-1))
# Split the targets into training/testing sets
linnerud_y_train = Y
#linnerud_y_test #= Y[-10:]
print(linnerud_y_train)
#print(linnerud_y_test)
# Create linear regression object
regr = linear_model.LinearRegression()
# Train the model using the training sets
regr.fit(linnerud_X_train,linnerud_y_train)
# Make predictions using the testing set
linnerud_y_pred = regr.predict(linnerud_X_test)
我期望在以下示例中实现类似的结果, https://scikit-learn.org/stable/auto_examples/linear_model/plot_ols.html