自定义数据集使用scikit-learn中的包实现线性回归方法对其进行拟合

代码

python 复制代码
import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

# 自定义数据集
data = np.array([
    [0.8, 1.0],
    [1.7, 0.9],
    [2.7, 2.4],
    [3.2, 2.9],
    [3.7, 2.8],
    [4.2, 3.8],
    [4.2, 2.7]
])

# 提取特征(X)和目标值(y)
X = data[:, 0].reshape(-1, 1)  # 特征值(二维数组)
y = data[:, 1]  # 目标值

# 创建并训练线性回归模型
model = LinearRegression()
model.fit(X, y)

# 生成预测值
y_pred = model.predict(X)

# 可视化数据和拟合结果

plt.figure('show figure')
plt.xlim(0, 5)
plt.ylim(0, 5)

# 绘制原始数据点
plt.scatter(X, y, color='blue')

# 绘制回归直线
plt.plot(X, y_pred, color='red')

# 添加标题和标签
plt.xlabel("X")
plt.ylabel("y")

# 显示图例
plt.legend()

# 显示图形
plt.show()

实验结果

相关推荐
用户8356290780511 小时前
从手动编辑到代码生成:Python 助你高效创建 Word 文档
后端·python
侃侃_天下1 小时前
最终的信号类
开发语言·c++·算法
c8i2 小时前
python中类的基本结构、特殊属性于MRO理解
python
echoarts2 小时前
Rayon Rust中的数据并行库入门教程
开发语言·其他·算法·rust
liwulin05062 小时前
【ESP32-CAM】HELLO WORLD
python
Aomnitrix2 小时前
知识管理新范式——cpolar+Wiki.js打造企业级分布式知识库
开发语言·javascript·分布式
Doris_20232 小时前
Python条件判断语句 if、elif 、else
前端·后端·python
Doris_20233 小时前
Python 模式匹配match case
前端·后端·python
每天回答3个问题3 小时前
UE5C++编译遇到MSB3073
开发语言·c++·ue5
伍哥的传说3 小时前
Vite Plugin PWA – 零配置构建现代渐进式Web应用
开发语言·前端·javascript·web app·pwa·service worker·workbox