基于文本的问题生成(五)

五折交叉验证

交叉验证法
交叉验证(cross-validation 简称cv)将数据集分为k等份,对于每一份数据集,其中k-1份用作训练集,单独的那一份用作验证集。一般采用xgboost.cv可以进行交叉验证

        for i, param in enumerate(param_grid):
            cv_result = xgb.cv(param, self.train_matrix,
                               num_boost_round=self.num_boost_round,  # max iter round
                               nfold=self.nfold,
                               stratified=self.stratified,
                               metrics=self.metrics,  # metrics focus on
                               early_stopping_rounds=self.early_stopping_rounds)  # stop when metrics not get better
            cur_auc = cv_result.iloc[len(cv_result)-1, 0]
            cur_iter_round = len(cv_result)
            if cur_auc > best_auc:
                best_auc, best_param, best_iter_round = cur_auc, param, cur_iter_round
            print('Param select {}, auc: {}, iter_round: {}, params: {}, now best auc: {}'
                  .format(i, cur_auc, cur_iter_round, param, best_auc))

输出结果

E:\PycharmProject\interrogative\venv\Scripts\python.exe E:/PycharmProject/interrogative/manage.py
Building prefix dict from the default dictionary ...
Loading model from cache C:\Users\dell\AppData\Local\Temp\jieba.cache
Loading model cost 0.596 seconds.
Prefix dict has been built successfully.

Param select 0, auc: 0.9873519999999999, iter_round: 270, params: {'eta': 0.1, 'max_depth': 4, 'objective': 'binary:logistic', 'silent': 0, 'subsample': 0.5}, now best auc: 0.9873519999999999



Param select 1
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值