python一元线性回归怎么实现

python

小亿

217

2023-08-18 06:26:55

栏目: 编程语言

在Python中，可以使用NumPy和Scikit-learn库来实现一元线性回归。

以下是一元线性回归的实现步骤：

导入所需的库：

import numpy as np
from sklearn.linear_model import LinearRegression

定义输入特征X和目标变量y：

X = np.array([[1], [2], [3], [4]])  # 输入特征
y = np.array([2, 4, 5, 8])  # 目标变量

创建线性回归模型对象：

model = LinearRegression()

将数据拟合到模型中：

model.fit(X, y)

预测新的数据：

X_new = np.array([[5]])  # 新输入特征
y_pred = model.predict(X_new)

完整的代码如下所示：

import numpy as np
from sklearn.linear_model import LinearRegression
# 定义输入特征X和目标变量y
X = np.array([[1], [2], [3], [4]])
y = np.array([2, 4, 5, 8])
# 创建线性回归模型对象
model = LinearRegression()
# 将数据拟合到模型中
model.fit(X, y)
# 预测新的数据
X_new = np.array([[5]])
y_pred = model.predict(X_new)
print(y_pred)

运行代码后，将输出预测结果。这就是一元线性回归的实现方法。

python一元线性回归怎么实现

最新问答

相关标签