Compatibility between outputs of skrub.datasets.fetch_X
and sklearn.datasets.fetch_X
#701
baggiponte
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Currently, sklearn datasets return a
sklearn.utils._bunch.Bunch
with the following attributes:['DESCR', 'data', 'feature_names', 'frame', 'target', 'target_names']
.skrub
data objects return askrub.datasets._fetching.DatasetAll
instead with the following attributes:(huge fan of dataclasses by the way)
This works fine but I was wondering if you wanted to be consistent with the
sklearn
API. The biggest con is that it might be a bit tedious to rewrite stuff. But if you inherit from the Bunch object you'd have less stuff to maintain too, I guess?Beta Was this translation helpful? Give feedback.
All reactions