Skip to content

Commit

Permalink
enable categorical in xgboost
Browse files Browse the repository at this point in the history
that lead to a test fail when non-cyclical encodings are used
  • Loading branch information
MoritzM00 committed Jan 8, 2025
1 parent b679bf4 commit a07945c
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/probafcst/models/xgboost.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ def __init__(
model = xgb.XGBRegressor(
objective="reg:quantileerror",
quantile_alpha=quantiles,
enable_categorical=True,
**self.kwargs,
)
super().__init__(
Expand Down

0 comments on commit a07945c

Please sign in to comment.