怎么做到训练前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))
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
悬赏问题
- ¥170 如图所示配置eNSP
- ¥20 docker里部署springboot项目,访问不到扬声器
- ¥15 netty整合springboot之后自动重连失效
- ¥15 悬赏!微信开发者工具报错,求帮改
- ¥20 wireshark抓不到vlan
- ¥20 关于#stm32#的问题:需要指导自动酸碱滴定仪的原理图程序代码及仿真
- ¥20 设计一款异域新娘的视频相亲软件需要哪些技术支持
- ¥15 stata安慰剂检验作图但是真实值不出现在图上
- ¥15 c程序不知道为什么得不到结果
- ¥15 键盘指令混乱情况下的启动盘系统重装