我正在研究多项式训练测试拟合问题,想把一个列表对象转换成一个numpy数组(4100)。(即4行,100列)
我有以下代码:
from sklearn.linear_model import LinearRegression
from sklearn.preprocessing import PolynomialFeatures
from numpy import array
np.random.seed(0)
n = 15
x = np.linspace(0,10,n) + np.random.randn(n)/5
y = np.sin(x)+x/6 + np.random.randn(n)/10
X_train, X_test, y_train, y_test = train_test_split(x, y, random_state=0)
results = []
pred_data = np.linspace(0,10,100)
degree = [1,3,6,9]
y_train1 = y_train.reshape(-1,1)
for i in degree:
poly = PolynomialFeatures(degree=i)
pred_poly1 = poly.fit_transform(pred_data[:,np.newaxis])
X_F1_poly = poly.fit_transform(X_train[:,np.newaxis])
linreg = LinearRegression().fit(X_F1_poly, y_train1)
pred = linreg.predict(pred_poly1)
results.append(pred)
dataArray = np.array(results).reshape(4, 100)
return dataArray
代码运行良好并返回一个数组(4100),但是输出看起来像是100行4列的内容,一旦我从np.数组函数,输出的维数变为(4100,1)。(我为我的无知道歉,1 in(4100,1)代表什么?)在
我想我对清单的理解有点问题,我现在还没弄明白。有谁能帮我指出代码中的错误,或者建议如何将输出数组转换成所需的(4100)格式?在
谢谢。在