使用sklearn进行编程

时间:2018-08-05 16:23:12

标签: python scikit-learn

请我在sklearn中使用预测技术 我有这个文件:

x | y
1| 1
2| 4
4|16

->预测技术必须给我这个模型y = x * x

使用该模型,我可以预测x = 3的值y 这是一个示例示例。在我的实验中,我使用了1000行的文件。 拜托,我该如何在代码上转换这些行?

 from sklearn.preprocessing import PolynomialFeatures 
 from sklearn import linear_model  
 X = [[1, 2, 4], [1, 4, 16]]  
 poly = PolynomialFeatures(degree=2) 
 X_ = poly.fit_transform(X)

1 个答案:

答案 0 :(得分:0)

也许您应该尝试这样的事情

import numpy as np
from  sklearn import linear_model

X=np.array([[1, 2, 4]]).T**2
y=np.array([1, 4, 16])

model=linear_model.LinearRegression()
model.fit(X,y)
X_predict=np.array([[3,1]]).T**2
model.predict(X_predict)