在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)
运行代码后,将输出预测结果。这就是一元线性回归的实现方法。