从零开始：构建你的第一个机器学习模型_适合从0开始训练的小模型-CSDN博客

本文链接：https://blog.csdn.net/weixin_44383927/article/details/140513424

在机器学习的浩瀚宇宙中，迈出第一步总是充满挑战又激动人心的。本文旨在通过一个简单而经典的案例——线性回归，引领你动手构建首个机器学习模型，让你从零开始，逐步掌握模型构建的基本流程。

一、理论基础：线性回归简介

线性回归是最基本的机器学习算法之一，用于预测连续值输出。它假设输入特征与输出之间存在线性关系，目标是找到一条直线（或多维空间中的超平面），使得所有数据点到直线的偏差平方和最小。

二、准备工具与数据

我们将使用Python语言，配合两个强大的库：numpy用于数学运算，sklearn（Scikit-learn）来构建和评估模型。

首先，安装必要的库（如果你尚未安装）：

pip install numpy scikit-learn

三、实战演练：构建线性回归模型

1. 数据准备

假设我们有一组关于房屋面积与价格的数据，目标是根据房屋面积预测价格。这里为了简化，我们构造一些示例数据。

import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error

# 构造样本数据
np.random.seed(0)
X = np.random.rand(100, 1) * 200  # 房屋面积
y = 50 * X + np.random.randn(100, 1) * 10  # 价格，假设每平方米50元，加入随机噪声

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

2. 模型训练

接下来，使用LinearRegression类来创建模型，并用训练数据拟合模型。

model = LinearRegression()
model.fit(X_train, y_train)

3. 预测与评估

模型训练完成后，我们可以在测试集上进行预测，并评估模型的表现。

predictions = model.predict(X_test)
mse = mean_squared_error(y_test, predictions)
print(f"Mean Squared Error: {mse:.2f}")

四、拓展：决策树模型

除了线性回归，决策树是另一个直观且易于理解的模型，适用于分类和回归任务。让我们快速构建一个简单的决策树模型，同样以预测房价为例。

from sklearn.tree import DecisionTreeRegressor

# 使用决策树回归模型
tree_model = DecisionTreeRegressor(random_state=42)
tree_model.fit(X_train, y_train.ravel())  # 注意ravel()用于展平一维数组

tree_predictions = tree_model.predict(X_test)
tree_mse = mean_squared_error(y_test, tree_predictions)
print(f"Decision Tree Mean Squared Error: {tree_mse:.2f}")