下面是我训练的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
目前没有回答
相关问题 更多 >
编程相关推荐