划分数据集的代码
时间: 2023-07-09 11:16:08 浏览: 85
好的,以下是一个简单的 Python 代码,用于将数据集划分成训练集和测试集:
```python
import random
def split_dataset(data, split_ratio):
train_size = int(len(data) * split_ratio)
train_data = []
test_data = list(data)
while len(train_data) < train_size:
index = random.randrange(len(test_data))
train_data.append(test_data.pop(index))
return train_data, test_data
```
其中,`data` 是原始的数据集,`split_ratio` 是训练集所占比例。函数返回训练集和测试集两个列表,可以根据需要进行使用。
阅读全文