python实线线性回归
时间: 2023-11-05 21:04:33 浏览: 71
python机器学习算法实训 – (一) 线性回归
5星 · 资源好评率100%
可以使用Python中的线性回归模型来进行实现。线性回归模型的最大优势是具有可解释性,权重有易于理解的解释,并且带有置信区间的估计。可以通过以下步骤来实现线性回归模型:
1. 导入所需的库:首先,需要导入所需的库,包括`numpy`、`pandas`和`sklearn`等。
2. 准备数据:将数据集分为特征和目标变量,并将其分成训练集和测试集。
3. 创建模型:使用`sklearn`库中的`LinearRegression`类创建线性回归模型。
4. 拟合模型:使用训练集数据拟合线性回归模型。
5. 预测结果:使用测试集数据进行预测。
6. 评估模型:使用评估指标(如均方误差、决定系数等)评估模型的性能。
下面是一个示例代码,展示了如何实现简单的线性回归模型:
```
import numpy as np
import pandas as pd
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error, r2_score
# 准备数据
X_train = np.array([[1], [2], [3], [4], [5]]) # 特征变量
y_train = np.array([2, 4, 6, 8, 10]) # 目标变量
# 创建模型
model = LinearRegression()
# 拟合模型
model.fit(X_train, y_train)
# 预测结果
X_test = np.array([[6], [7], [8], [9], [10]]) # 测试集特征变量
y_pred = model.predict(X_test) # 预测结果
# 评估模型
mse = mean_squared_error(y_train, model.predict(X_train))
r2 = r2_score(y_train, model.predict(X_train))
# 打印结果
print("预测结果:", y_pred)
print("均方误差:", mse)
print("决定系数:", r2)
```
阅读全文