怎么做到训练前90%行并测试最后10%行?在test.size和train.size可以设置吗?
1条回答 默认 最新
- heart_6662 2022-12-28 18:35关注
望采纳!点击该回答右侧的“采纳”按钮即可采纳!!
可以使用pandas的sample方法来随机选择数据的行,然后使用train_test_split函数将数据分为训练集和测试集。代码我给你写了一份,有用可以采纳博主的答案呀
import pandas as pd from sklearn.model_selection import train_test_split 读取数据 data = pd.read_csv('data.csv') 随机选择数据的行 data_sample = data.sample(frac=1) 取前90%的数据作为训练集,剩余的数据作为测试集 train_size = int(len(data_sample) * 0.9) train = data_sample[:train_size] test = data_sample[train_size:] 也可以使用train_test_split函数将数据分为训练集和测试集 X = data_sample.iloc[:, :-1] y = data_sample.iloc[:, -1] X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1, random_state=42) print('训练集大小:', len(X_train)) print('测试集大小:', len(X_test))
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
悬赏问题
- ¥15 装 pytorch 的时候出了好多问题,遇到这种情况怎么处理?
- ¥20 IOS游览器某宝手机网页版自动立即购买JavaScript脚本
- ¥15 手机接入宽带网线,如何释放宽带全部速度
- ¥30 关于#r语言#的问题:如何对R语言中mfgarch包中构建的garch-midas模型进行样本内长期波动率预测和样本外长期波动率预测
- ¥15 ETLCloud 处理json多层级问题
- ¥15 matlab中使用gurobi时报错
- ¥15 这个主板怎么能扩出一两个sata口
- ¥15 不是,这到底错哪儿了😭
- ¥15 2020长安杯与连接网探
- ¥15 关于#matlab#的问题:在模糊控制器中选出线路信息,在simulink中根据线路信息生成速度时间目标曲线(初速度为20m/s,15秒后减为0的速度时间图像)我想问线路信息是什么