Linux 拨号vps windows公众号手机端

Scikit-learn中怎么实现线性回归

lewis 1年前 (2024-04-05) 阅读数 13 #程序编程
文章标签 Scikit-learn

在Scikit-learn中,可以使用LinearRegression类来实现线性回归。下面是一个简单的示例代码:

from sklearn.linear_model import LinearRegression
import numpy as np

# 创建一些示例数据
X = np.array([[1], [2], [3], [4]])
y = np.array([2, 4, 6, 8])

# 创建线性回归模型
model = LinearRegression()

# 拟合模型
model.fit(X, y)

# 进行预测
y_pred = model.predict([[5]])

print(y_pred)

在这个示例中,首先创建了一些示例数据X和y,然后创建了一个LinearRegression模型,并用fit方法拟合了这些数据。最后使用predict方法对新的数据进行预测。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门