衡量适合多项式的模型

时间:2018-11-25 06:37:47

标签: python coefficients

我正在研究波士顿房屋价格数据集。我正在尝试确定多项式模型的模型拟合(确定系数)。

X = boston_data["lstat"].values.reshape(-1, 1)
y = boston_data["medv"].values.reshape(-1, 1)

# Transform X into polynomial matrix
poly_2_x = PolynomialFeatures(degree=2).fit_transform(X)

# Fit data to model
poly_2_reg = LinearRegression().fit(poly_2_x, y)

# Calculate coefficient
poly_2_coff = poly_2_reg.score(poly_2_x, y)

但是,我遇到以下错误: ValueError:找到输入样本数量不一致的输入变量:[333,6]

是否有更好的方法?还是应该更改以纠正错误?

谢谢!

0 个答案:

没有答案