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

   2024-10-19 3590
核心提示:在Scikit-learn中,可以使用LinearRegression类来实现线性回归。下面是一个简单的示例代码:from sklearn.linear_model import L

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

from sklearn.linear_model import LinearRegressionimport 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方法对新的数据进行预测。

 
举报打赏
 
更多>同类维修大全
推荐图文
推荐维修大全
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号