MLBazaar/BTB

OneHotEncoder `TypeError: __init__() got an unexpected keyword argument 'sparse_output'`

Opened this issue · 0 comments

sarahmish commented
  • baytune 0.5.0
  • macOS 14.1
  • python 3.8

The following line produces an error when scikit-learn<1.2. I would suggest to add it as a requirement for the package.

BTB/baytune/tuning/hyperparams/categorical.py

Line 52 in 9b4500e

self._encoder = OneHotEncoder(categories=[choices], sparse_output=False)

reference:

New in version 1.2: sparse was renamed to sparse_output

https://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.OneHotEncoder.html