多元线性回归模型的Python实现
时间: 2023-09-03 14:12:54 浏览: 108
python实现机器学习之多元线性回归
5星 · 资源好评率100%
多元线性回归模型是一种可以用于预测多个自变量对因变量的影响的模型。在Python中,可以使用Scikit-learn库来实现多元线性回归模型。
以下是一个简单的例子,展示了如何使用Scikit-learn库来实现多元线性回归模型:
```python
# 导入需要的库
import numpy as np
from sklearn.linear_model import LinearRegression
# 准备数据
X = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
y = np.array([6, 15, 24, 33])
# 创建模型并训练
model = LinearRegression()
model.fit(X, y)
# 预测新数据
new_data = np.array([[13, 14, 15], [16, 17, 18]])
predictions = model.predict(new_data)
print(predictions)
```
在上面的例子中,我们首先导入了必要的库。然后,我们准备了一个包含自变量的矩阵X和一个包含因变量的向量y。接下来,我们创建了一个线性回归模型,并使用fit()方法来训练它。最后,我们使用predict()方法来预测新数据并打印出结果。
需要注意的是,多元线性回归模型的实现需要一些前置知识,比如线性代数和统计学。如果你对这些知识还不熟悉,建议先学习一下。
阅读全文