pycharm中输出数据集中训练集和测试集的代码
时间: 2023-05-21 08:05:32 浏览: 108
爬虫代码,数据集寻找加自动分类,训练集,测试集自动生成!下载就可用,python
你可以使用如下代码打印出数据集中的训练集和测试集文件路径:
```
import os
from sklearn.model_selection import train_test_split
# 定义数据集路径
data_path = '/path/to/data'
# 加载数据集
data = pd.read_csv(os.path.join(data_path, 'data.csv'))
# 划分训练集和测试集,设置测试集比例为0.2
train_data, test_data = train_test_split(data, test_size=0.2, random_state=42)
# 打印训练数据集和测试数据集路径
print('训练集路径:', os.path.join(data_path, 'train.csv'))
print('测试集路径:', os.path.join(data_path, 'test.csv'))
```
请注意,上述代码需要导入 pandas, os 和 sklearn。
阅读全文