我可以在培训xgb型号时打印时间吗?

2024-09-29 19:35:24 发布

您现在位置:Python中文网/ 问答频道 /正文

下面是我训练的xgboost模型的代码和结果。我想知道训练每个迭代需要多长时间,以便更好地估计我以后可以训练多少轮。是否有一种方法可以打印出每轮后模型训练所用的时间,例如,在与结果相同的行中? 谢谢大家!

import xgboost as xgb
dtrain = xgb.DMatrix(X_train, label=y_train)

# parameters
param = {'max_depth': 2, 'eta': 1, 'objective': 'binary:logistic'}
param['nthread'] = 4
param['eval_metric'] = 'auc'
evallist = [(dtest, 'eval'), (dtrain, 'train')]

num_round = 10
bst = xgb.train(param, dtrain, num_round, evallist)

结果:

[0] eval-auc:0.618416   train-auc:0.621827
[1] eval-auc:0.733406   train-auc:0.734433
...
[8] eval-auc:0.854423   train-auc:0.856366
[9] eval-auc:0.855932   train-auc:0.857541

Tags: 方法代码模型parameval时间trainnum

热门问题