Chryzanthemum / xgb2sql

Apache License 2.0
41 stars 3 forks source link

Unable to Specify Gamma Parameter #16

Open acharabin opened 1 month ago

acharabin commented 1 month ago

When instantiating the XGBoost estimator using a custom gamma parameter, the following error is thrown:

TypeError: 'dict_values' object is not subscriptable

acharabin commented 1 month ago

After digging further, the issue doesn't appear to be directly related to the gamma parameter, but rather the indirect consequence of adding certain values of a gamma parameter that may result in a tree's leaf length of 1.

The proposed code change can be found in this pull request - https://github.com/Chryzanthemum/xgb2sql/pull/17/commits/0fecf4752808d0aac370238a5e4233efca028ca8