利用train_test_split函数把两个列表数据分解为训练和测试数据的代码示例
时间: 2023-11-02 16:01:52 浏览: 36
下面是一个利用`train_test_split`函数将两个列表数据分解为训练和测试数据的代码示例:
```python
from sklearn.model_selection import train_test_split
# 假设有两个列表数据,分别为x和y
x = [1, 2, 3, 4, 5]
y = [10, 20, 30, 40, 50]
# 利用train_test_split函数将x和y分解为训练和测试数据,其中test_size表示测试数据的比例
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2)
# 打印分解后的训练和测试数据
print("训练数据:")
print("x_train:", x_train)
print("y_train:", y_train)
print("测试数据:")
print("x_test:", x_test)
print("y_test:", y_test)
```
输出结果如下:
```
训练数据:
x_train: [4, 5, 3, 1]
y_train: [40, 50, 30, 10]
测试数据:
x_test: [2]
y_test: [20]
```
在本例中,我们将`x`和`y`列表数据按照`test_size=0.2`的比例划分为训练和测试数据。训练数据包含80%的数据,测试数据包含20%的数据。`train_test_split`函数会自动随机划分数据。最终,我们得到了`x_train`, `y_train`, `x_test`和`y_test`四个列表,分别表示训练数据和测试数据的输入和输出。
相关推荐
![py](https://img-home.csdnimg.cn/images/20210720083646.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)