时间戳

首页 > 解决方案 > 是否有 python 函数可以为 LightGBM 迭代添加进度条？

python - 是否有 python 函数可以为 LightGBM 迭代添加进度条？

问题描述

我正在尝试为 LGBMRegressor 拟合"num_iterations": 10000，单元格输出非常长，迭代次数为 10000 次。我想添加一个进度条来跟踪迭代次数。我正在使用下面的代码来适应LGBMRegressor

gbm = lgb.LGBMRegressor(**hyper_params)

gbm.fit(X_train, y_train,
        eval_set=[(X_test, y_test)],
        eval_metric='mae',
        early_stopping_rounds=1000)

输出：

[1] valid_0's l1: 8.1618    valid_0's l2: 107.11
Training until validation scores don't improve for 1000 rounds
[2] valid_0's l1: 8.12118   valid_0's l2: 106.048
[3] valid_0's l1: 8.08075   valid_0's l2: 104.996
[4] valid_0's l1: 8.04053   valid_0's l2: 103.955



[9998]  valid_0's l1: 0.0344606 valid_0's l2: 0.00281218
[9999]  valid_0's l1: 0.0344581 valid_0's l2: 0.00281191
[10000] valid_0's l1: 0.0344559 valid_0's l2: 0.00281167
Did not meet early stopping. Best iteration is:
[10000] valid_0's l1: 0.0344559 valid_0's l2: 0.00281167

LGBMRegressor(bagging_fraction=0.7, bagging_freq=10, feature_fraction=0.9,
              learning_rate=0.005, max_bin=512, max_depth=8,
              metric=['mae', 'mse'], n_estimators=1000, num_iterations=10000,
              num_leaves=128, objective='regression', task='train', verbose=0)

上面只添加了几个迭代。

到目前为止我没有找到任何添加进度条的东西。是否可以为迭代添加进度条，有或没有 valid_0's l1 valid_0's l2

是否有 python 代码为 lightgbm 迭代添加像 tqdm 这样的进度条？

标签： pythonmachine-learningprogress-barlightgbmtqdm

解决方案

推荐阅读