多项式线性回归中系数的阶数

2024-10-04 09:19:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用sklearn在Python中构建一个简单的多项式模型。我的训练集包含3个特征向量(即RGB三元组)。我正在使用以下代码装配模型:

degree = 1
model = make_pipeline(PolynomialFeatures(degree), Ridge())
model.fit(samples, y)

coef = model.named_steps['ridge'].coef_ 
intercept = model.named_steps['ridge'].intercept

用一次多项式很好。这是输出:

^{pr2}$

为了用我刚刚训练的模型评估一个新的三元组,我可以利用:

prediction = model.predict(aNewTriplet)

或者,我可以:

prediction = intercept + (-0.123578 * aNewTriplet[0] + 0.8367021 * aNewTriplet[1] - 1.71648988 * aNewTriplet[2])

如果我有一个二阶多项式(它输出10个元素的数组)呢?系数如何排序?这样地?公司名称:

coefficients (order = 2) = [ 0, R, G, B, R^2, G^2, B^2, RG, RB, GR]

Tags: 代码模型modelrgbsklearnstepsnamed三元组