怎么设置train_test_split不随机
时间: 2023-10-13 13:22:45 浏览: 49
您可以通过设置随机种子来确保 train_test_split 不随机,例如:
```python
from sklearn.model_selection import train_test_split
random_state = 42 # 设置随机种子
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=random_state)
```
在这个例子中,train_test_split 将根据随机种子(set.seed)设置来将数据集分成训练集和测试集,这将确保每次运行代码都会得到相同的训练集和测试集。
相关问题
train_test_split怎么设置不随机
train_test_split函数可以通过设置random_state参数来实现不随机的划分。当random_state参数被设置为一个固定的值时,每次运行train_test_split函数时,都会得到相同的划分结果。
以下是一个示例代码:
```python
from sklearn.model_selection import train_test_split
# 假设X和y是你的数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
```
在上述代码中,random_state参数被设置为42。无论何时运行这段代码,都会得到相同的训练集和测试集划分结果。
train_test_split函数不随机划分
train_test_split函数默认是随机划分训练集和测试集的。如果你想要固定划分结果,可以使用参数random_state来设置随机种子。下面是一个示例:
```python
from sklearn.model_selection import train_test_split
# 假设train_data和train_target是你的训练数据和目标变量
X_train, X_test, y_train, y_test = train_test_split(train_data, train_target, test_size=0.4, random_state=0)
# 输出划分后的训练集和测试集的大小
print("训练集大小:", len(X_train))
print("测试集大小:", len(X_test))
```
这个例子中,train_data和train_target是你的训练数据和目标变量。train_test_split函数将会随机将数据划分为训练集和测试集,其中测试集的大小占总数据的40%。设置random_state为0保证了每次运行代码时得到的划分结果是一致的。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pptx](https://img-home.csdnimg.cn/images/20210720083543.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)