线性回归模型python
时间: 2023-11-03 07:06:27 浏览: 129
python建模算法 - 线性回归.rar
线性回归模型的python实现通常使用scikit-learn库中的线性回归算法。在使用该库之前,你需要先安装该库并导入相应的模块。下面是一个简单的线性回归模型的python代码示例:
```python
from sklearn.linear_model import LinearRegression
# 假设我们有一些输入特征X和对应的输出Y
X = [[1], [2], [3], [4]]
Y = [2, 4, 6, 8]
# 创建线性回归模型对象
model = LinearRegression()
# 使用训练数据拟合模型
model.fit(X, Y)
# 使用模型进行预测
x_test = [[5], [6]]
predictions = model.predict(x_test)
# 打印预测结果
for i, x in enumerate(x_test):
print("预测结果:", predictions[i], "输入:", x)
```
这段代码首先导入了LinearRegression类,然后创建了一个线性回归模型对象model。接下来,我们使用fit()方法拟合模型,传入输入特征X和对应的输出Y。然后,我们使用predict()方法对新的输入特征进行预测,并打印预测结果。
请注意,这只是一个简单的示例,实际应用中可能会涉及更多的特征和更复杂的数据。你可以根据具体的数据和需求进行适当的调整和扩展。
阅读全文