10分钟掌握机器学习：从零开始搭建你的第一个预测模型！

程序员一粟

于 2025-04-19 13:46:07 发布

阅读量594

点赞数 9

文章标签：机器学习人工智能数据库开发语言 ai

本文链接：https://blog.csdn.net/jennycisp/article/details/147349810

版权

引言

在魔搭社区，我们可以通过低代码或无代码的方式快速搭建并部署机器学习模型。这篇文章将详细介绍如何通过 Python 构建一个完整的机器学习工作流，包括数据加载、可视化、模型训练、预测和评估等步骤。

一、加载数据并初步探索

机器学习的第一步是加载数据并了解其基本结构。这可以帮助我们快速识别数据中的特征和潜在问题。导入一个csv文件到虚拟机上，这里我们使用 Pandas 读取数据，并查看前几行内容。

代码示例：

ini 代码解读复制代码# python 模块引入
# python 最流行的数据分析库
import pandas as pd
# js 异步的 IO 操作
# python 同步
df_ads = pd.read_csv("直播带货.csv")
# 数据样本的尺寸
# print(df_ads.size)
# 不传参数默认返回五条
df_ads.head(10)

在这里插入图片描述
目的：

了解数据结构，包括列名、数据类型和样本大小。
确定目标值（成交额）和特征（如转发量）。

二、数据可视化：直观理解数据

通过数据可视化，我们可以更清晰地观察特征之间的关系。例如，分析“转发量”和“成交额”的相关性。

代码示例：

dart 代码解读复制代码import matplotlib.pyplot as plt 

plt.plot(df_ads['转发量'], df_ads['成交额'], 'r.', label='data dot') 
plt.xlabel('shares') 
plt.ylabel('sales') 
plt.legend() 
plt.show()

在这里插入图片描述
目的：

观察特征间是否存在线性关系，为选择模型提供参考。

三、数据清洗与预处理

分离特征和目标值

ini 代码解读复制代码# 移除成交额这一列 
# 影响成交额的影响因子拿出来 训练
X = df_ads.drop(["成交额"], axis = 1)
# X 
# 成交额 特征
y = df_ads.成交额
print(X.head())
y.head()

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传 目的：

确保数据完整、格式正确，为模型训练做好准备。

四、划分训练集和测试集

为了验证模型的泛化能力，我们将数据分为训练集（80%）和测试集（20%）。

代码示例：

ini 代码解读复制代码# 数据训练
# 简单模型，线性回归
from sklearn.linear_model import LinearRegression
# 线性回归算法的模型实例
model = LinearRegression()

# 训练 测试
from sklearn.model_selection import train_test_split
# train_test_split 将数据分成 训练数据，测试部分
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)

X_train.size
X_test.size

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

目的：

训练集用于训练模型，测试集用于评估模型的性能。

五、构建模型并训练

我们选择线性回归模型，这是最简单的机器学习模型之一，用于预测成交额。

代码示例：

bash 代码解读复制代码# 训练 y= ax + b 趋势 
model.fit(X_train, y_train)

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传 目的：

通过拟合训练集数据，构建预测模型。

拓展之线性回归

线性回归对已有数据进行建模，可以对未来数据进行预测。线性回归是所有机器学习技术的一个最好起点，很多复杂的机器学习技术以及当前大火的深度神经网络都或多或少基于线性回归。上述使用的是一元线性回归方程 y = ax + b。

六、预测与结果展示

训练完成后，我们使用测试集进行预测，并将真实值与预测值进行对比。

代码示例：

scss 代码解读复制代码# 拷贝测试集数据
df_ads_pred = X_test.copy() 
df_ads_pred.head()
df_ads_pred['成交额真值'] = y_test
df_ads_pred['成交额预测值'] = y_pred
df_ads_pred

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传 目的：

检查模型对未知数据的预测能力。

七、模型评估

通过评价指标，量化模型的性能。常用指标包括模型评分（R²）和均方误差（MSE）。

代码示例：

bash 代码解读复制代码# 大模型自己打分
# 预期的结果数据 衡量模型的训练质量 
y_pred = model.predict(X_test)

print("线性回归预测集评分", model.score(X_test, y_test))
print("线性回归训练集评分", model.score(X_train, y_train))

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传 目的：

确定模型对训练集和测试集的拟合程度。

八、可视化预测结果

通过图表展示真实值与预测值的对比，直观了解模型的表现。

代码示例：

ini 代码解读复制代码X = df_ads[['转发量']]
X.head()
y = df_ads.成交额
y.head()

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)
model = LinearRegression()
model.fit(X_train, y_train)

y_pred = model.predict(X_test)
# 画布 大小是 10x6
plt.figure(figsize=(10, 6))
# 散点图
plt.scatter(X_test, y_test, color='red',label="true value")
plt.plot(X_test, y_pred, color='red', 
         linewidth=2, label="predict value")
plt.xlabel("shares")
plt.ylabel("selles")
plt.title('shares VS selles')
plt.legend()
plt.grid(True)
plt.show()

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传 目的：