自定义数据集使用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()

实验结果

相关推荐
黄雪超34 分钟前
JVM——函数式语法糖:如何使用Function、Stream来编写函数式程序?
java·开发语言·jvm
ThetaarSofVenice41 分钟前
对象的finalization机制Test
java·开发语言·jvm
思则变1 小时前
[Pytest] [Part 2]增加 log功能
开发语言·python·pytest
lijingguang1 小时前
在C#中根据URL下载文件并保存到本地,可以使用以下方法(推荐使用现代异步方式)
开发语言·c#
漫谈网络1 小时前
WebSocket 在前后端的完整使用流程
javascript·python·websocket
¥-oriented2 小时前
【C#中路径相关的概念】
开发语言·c#
CoderCodingNo2 小时前
【GESP】C++四级考试大纲知识点梳理, (7) 排序算法基本概念
开发语言·c++·排序算法
恋猫de小郭2 小时前
Meta 宣布加入 Kotlin 基金会,将为 Kotlin 和 Android 生态提供全新支持
android·开发语言·ios·kotlin
JosieBook2 小时前
【Java编程动手学】使用IDEA创建第一个HelloJava程序
java·开发语言·intellij-idea
Thomas_YXQ2 小时前
Unity3D DOTS场景流式加载技术
java·开发语言·unity