# 导入必要的库
from sklearn.linear_model import LinearRegression
import numpy as np
import matplotlib.pyplot as plt
# 创建一些示例数据
X = np.array([[1], [2], [3], [4], [5]]) # 特征
y = np.array([2, 4, 6, 8, 10]) # 目标变量
# 创建线性回归模型
model = LinearRegression()
# 训练模型
model.fit(X, y)
# 打印模型的系数和截距
print(f"系数: {model.coef_}")
print(f"截距: {model.intercept_}")
# 预测新数据点
X_new = np.array([[6]])
y_pred = model.predict(X_new)
print(f"预测值: {y_pred}")
# 可视化结果
plt.scatter(X, y, color='blue', label='真实值')
plt.plot(X, model.predict(X), color='red', label='预测值')
plt.scatter(X_new, y_pred, color='green', label='新预测点')
plt.legend()
plt.show()
LinearRegression
模型、numpy
用于处理数组,以及 matplotlib
用于可视化。X
是特征(输入),y
是目标变量(输出)。LinearRegression()
创建一个线性回归模型,并通过 fit()
方法训练模型。matplotlib
绘制散点图和拟合直线,展示模型的效果。上一篇:python创建线程
下一篇:python onnx
Laravel PHP 深圳智简公司。版权所有©2023-2043 LaravelPHP 粤ICP备2021048745号-3
Laravel 中文站