如何根据用户输入的值进行预测?

时间:2017-03-31 20:03:10

标签: scikit-learn

我正在进行多项式回归。我可以在真实值上绘制它,但问题是,我想给出一个X值,例如125,然后我想得到预测。

我该怎么做?

谢谢

1 个答案:

答案 0 :(得分:1)

我相信你正在寻找model.predict方法。

以下是一个例子:

model = make_pipeline(PolynomialFeatures(), Ridge())
model.fit(X, y)
print(model.predict([[125]]))