R语言机器学习系列：决策树回归

R语言机器学习系列：决策树回归

决策树是一种常用的机器学习算法，可以用于回归问题。在本文中，我们将介绍如何使用R语言实现决策树回归，并提供相应的源代码。

首先，我们需要安装并加载rpart包，该包提供了决策树算法的实现。可以使用以下命令安装该包：

install.packages("rpart")
library(rpart)

接下来，我们将使用一个示例数据集来演示决策树回归的过程。假设我们有一个包含自变量（特征）和因变量（目标）的数据集。我们的目标是根据自变量的值来预测因变量的值。

让我们假设数据集的文件名为dataset.csv，并且数据集包含两个自变量（X1和X2）和一个因变量（Y）。我们可以使用以下代码将数据集加载到R中：

dataset <- read.csv("dataset.csv")

在数据集加载完成后，我们可以将数据集分为训练集和测试集。训练集用于构建决策树模型，而测试集用于评估模型的性能。

set.seed(123)  # 设