我正在使用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]
目前没有回答
相关问题 更多 >
编程相关推荐