用python写一个验证线性回归的程序
时间: 2023-09-29 14:08:06 浏览: 80
用Python编写自己的线性回归分析程序.pdf
5星 · 资源好评率100%
好的,下面是一个简单的用 Python 验证线性回归模型的程序示例:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
# 创建一个随机数据集
np.random.seed(0)
x = np.random.rand(100, 1)
y = 2 * x + np.random.rand(100, 1)
# 创建线性回归模型
model = LinearRegression()
# 训练模型
model.fit(x, y)
# 输出模型的参数
print('斜率:', model.coef_)
print('截距:', model.intercept_)
# 预测新数据
x_new = np.array([[0.5], [0.6], [0.7]])
y_new = model.predict(x_new)
print('新数据预测结果:', y_new)
```
该程序使用 NumPy 创建了一个随机数据集,然后使用 Scikit-learn 中的 `LinearRegression` 类创建了一个线性回归模型。接着训练模型并输出模型的参数,最后预测了一些新数据并输出预测结果。
请注意,此程序仅用于演示如何使用 Python 验证线性回归模型,实际应用中需要根据具体情况进行调整和优化。
阅读全文