java划分训练集和测试集_训练集、验证集和测试集

通常在训练机器学习模型的时候,我们会将数据划分为训练集、验证集和测试集。一般来说,训练集:验证集:测试集的划分比例为6:2:2。对原始数据进行三个集合的划分,是为了能够选出效果(可以理解为准确率)最好的、泛化能力最佳的模型。

训练集(Training Set)

作用是用来拟合模型,通过设置分类器的参数,训练分类模型。后续结合验证集作用时,会选出同一参数的不同取值,拟合出多个分类器。

验证集(Cross ValidaDon Set)

作用是当通过训练集训练出多个模型后,为了能找出效果最佳的模型,使用各个模型对验证集数据进行预测,并记录模型准确率。

选出效果最佳的模型所对应的参数,即用来调整模型参数,如svn中的参数c和核函数等。在交叉验证过程中也需要设置验证集,如k-折交叉验证(k-fold crossValidation)。

测试集(Test Set)

通过训练集和验证集得出最优模型后,使用测试集进行模型预测,用来衡量该最优模型的性能和分类能力。即可以把测试集当做从来不存在的数据集,当已经确定模型参数后,使用测试集进行模型性能评价。

所以说,训练集(Training Set)是用来训练模型或确定模型参数的,如ANN中权值等; 验证集(Validation Set)是用来做模型选择(model selection),即做模型的最终优化及确定的,如ANN的结构;而测试集(Test Set)则纯粹是为了测试已经训练好的模型的泛化能力。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值