R语言随机划分训练组测试组
时间: 2023-07-24 16:04:01 浏览: 103
给定比例随机划分训练集、验证集和测试集
在R语言中,可以使用`sample()`函数来进行随机划分训练集和测试集。下面是一个示例代码:
```R
set.seed(1) # 设置随机数种子,保证每次运行结果一致
# 假设data是你要进行划分的数据集
# 随机生成索引
train_id <- sample(1:nrow(data), 0.85 * nrow(data))
test_id <- setdiff(1:nrow(data), train_id)
# 划分训练集和测试集
train <- data[train_id, ]
test <- data[test_id, ]
```
上述代码中,我们首先使用`sample()`函数随机生成了一个包含`0.85 * nrow(data)`个随机索引的向量`train_id`,表示训练集的索引。然后,使用`setdiff()`函数找到不在`train_id`中的索引,得到测试集的索引`test_id`。最后,根据索引将数据集划分为训练集和测试集。
你可以根据实际需求修改代码中的比例和变量名称。注意,设置随机数种子`set.seed()`可以保证每次运行结果一致,如果不设置种子,则每次运行代码将得到不同的随机划分结果。
阅读全文