第L2周:机器学习-线性回归

目标

  1. 学习简单线性回归模型和多元线性回归模型

  2. 通过代码实现:通过鸢尾花花瓣长度预测花瓣宽度
    具体实现
    (一)环境
    语言环境 :Python 3.10
    编 译 器: PyCharm
    框 架 :scikit-learn
    (二)具体步骤:
    造个数据集,内容格式如下:

    导入库

    import pandas as pd
    import numpy as np
    import matplotlib.pyplot as plt

简单线性回归

复制代码
# 加载数据
dataset = pd.read_csv('./studentscores.csv')  
print(dataset)
复制代码
# 取第一列Hours的值
X = dataset.iloc[:, :1].values  
print(X)
复制代码
# 取第二列Scores的值
Y = dataset.iloc[:, 1].values  
print(Y)

很好奇,看看X,Y的形状:

复制代码
print(X.shape)
print(Y.shape)

看来两者是一样的形状和大小 。继续:

复制代码
# 切分一下数据集,75%用来训练,25%用来测试
from sklearn.model_selection import  train_test_split  
X_train, X_test, Y_train, Y_test = train_test_split(X, Y,  
                                                    test_size=1/4,  
                                                    random_state=0)

检验一下切分的成果:

复制代码
print(X_train, X_train.shape)
复制代码
print(Y_train, Y_train.shape)
复制代码
print(X_test, X_test.shape)
print(Y_test, Y_test.shape)

做简单线性回归

复制代码
# 简单线性回归  
from sklearn.linear_model import LinearRegression  
  
regressor = LinearRegression()  
regressor = regressor.fit(X_train, Y_train)

# 预测一下结果  
Y_pred = regressor.predict(X_test)  
  
print(Y_pred, Y_pred.shape)

这个预测结果和上面的Y_test比较一下,可以看到两者之间的差距以及相似性。我们进行可视化直观看看:

复制代码
# 训练集可视化  
plt.scatter(X_train, Y_train, color='red')  
plt.plot(X_train, regressor.predict(X_train), color='blue')  
plt.show()

**

复制代码
# 测试集预测结果可视化  
plt.scatter(X_test, Y_test, color='red')  
plt.plot(X_test, regressor.predict(X_test), color='blue')  
plt.show()

红点是实际分布,蓝色线是预测趋势线。两者是趋于一致的,预测的偏离并不大。

注:plt.scatter()绘制散点图,plt.plot()绘制折线图。

下面看看多元线性回归,通过鸢尾花花瓣长度预测花瓣宽度

  1. 导入数据集

    url = 'https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data'
    names = ['花萼-length', '花萼-width', '花瓣-length', '花瓣-width', 'class']
    dataset = pd.read_csv(url, names=names)
    print(dataset)

  1. 分析一下数据

    plt.plot(dataset['花萼-length'], dataset['花瓣-width'], 'x', label="marker='x'")
    plt.plot(dataset['花萼-width'], dataset['花瓣-width'], 'o', label="marker='o'")
    plt.plot(dataset['花瓣-length'], dataset['花瓣-width'], 'v', label="marker='v'")
    plt.legend(numpoints=1)
    plt.show()

  1. 取数据

    取[花萼-width : 花瓣-length]

    X = dataset.iloc[:, [1, 2]].values
    print(X, X.shape)


复制代码
Y = dataset.iloc[:, 3].values  # 取花瓣-width值  
print(Y, Y.shape)
  1. 将dataset切分成训练数据集和测试数据集

    from sklearn.model_selection import train_test_split
    X_train, X_test, Y_train, Y_test = train_test_split(X, Y,
    test_size=0.2,
    random_state=0)

  2. 训练多元线性回归模型

    from sklearn.linear_model import LinearRegression
    regressor = LinearRegression()
    regressor.fit(X_train, Y_train)

  3. 在测试集上预测结果

    y_pred = regressor.predict(X_test)
    print(y_pred)

  1. 把测试集预测结果可视化

    plt.scatter(Y_test, y_pred, color='red')
    plt.plot(Y_test, Y_test, color='blue') # 假设预测100%正确,那么走势是蓝线
    plt.plot(Y_test)
    plt.xlabel("True")
    plt.ylabel("Prediction")
    plt.show()

相关推荐
AI小书房12 小时前
【人工智能通识专栏】第十三讲:图像处理
人工智能
fanstuck12 小时前
基于大模型的个性化推荐系统实现探索与应用
大数据·人工智能·语言模型·数据挖掘
多看书少吃饭13 小时前
基于 OpenCV 的眼球识别算法以及青光眼算法识别
人工智能·opencv·计算机视觉
IT学长编程13 小时前
计算机毕业设计 基于大数据技术的医疗数据分析与研究 Python 大数据毕业设计 Hadoop毕业设计选题【附源码+文档报告+安装调试】
大数据·hadoop·机器学习·数据分析·毕业设计·毕业论文·医疗数据分析
一条数据库13 小时前
南京方言数据集|300小时高质量自然对话音频|专业录音棚采集|方言语音识别模型训练|情感计算研究|方言保护文化遗产数字化|语音情感识别|方言对话系统开发
人工智能·音视频·语音识别
Yingjun Mo13 小时前
1. 统计推断-基于神经网络与Langevin扩散的自适应潜变量建模与优化
人工智能·神经网络·算法·机器学习·概率论
DogDaoDao14 小时前
Docker全解析:从核心概念到2025年AI集成新特性
人工智能·docker·eureka·程序员
深耕AI14 小时前
【PyTorch训练】准确率计算(代码片段拆解)
人工智能·pytorch·python
zskj_zhyl14 小时前
七彩喜智慧养老:用科技温暖晚年,让关爱永不掉线
人工智能·科技·机器人