博客

python使用sklearn实现线性回归

  • Post author:
  • Post category:python


from sklearn import linear_model
import numpy as np

model_reg = linear_model.LinearRegression()

# 训练数据集
data=[[1,2],[4,8]]
X_train = np.array(data)[:,0].reshape(-1,1)
y_train = np.array(data)[:,1].reshape(-1,1)

# 训练回归模型
model_reg.fit(X_train, y_train)
# 测试数据
test_data_X = np.array([10,12]).reshape(-1,1)
test_data_y = model_reg.predict(test_data_X)
print(test_data_y)

运行结果：

[[20.]

[24.]]

在这里插入图片描述


版权声明：本文为skyli114原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。