Documentation
搜索文档…
LightGBM
利用我们的回调函数,只需要一行代码就可以把LightGBM的性能可视化。
1
from wandb.lightgbm import wandb_callback
2
import lightgbm as lgb
3
4
....
5
6
gbm = lgb.train(params,
7
lgb_train,
8
num_boost_round=20,
9
valid_sets=lgb_eval,
10
valid_names=('validation'),
11
callbacks=[wandb_callback()])
Copied!
若要查看完整代码示例,请进入我们的示例仓库,或者用Colab笔记本。
最近更新 9mo ago
复制链接