Error when specifying sample_weight with xgboost

microsoft / FLAML

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

https://microsoft.github.io/FLAML/

MIT License

3.91k stars 508 forks source link

Error when specifying sample_weight with xgboost #53

Closed maycuiyan closed 3 years ago

maycuiyan commented 3 years ago

line 394 of flaml/model.py, The train() method does not accept weight as legitimate argument. Instead, the weight should be specified in line 390 when creating dtrain.

sonichi commented 3 years ago

Could you try version 0.2.10 and let me know if the problem is fixed? Thanks.

maycuiyan commented 3 years ago

@sonichi Upgrading to 0.2.10 solves the issue. Thanks! Will close this.