[ENH] Add load_classification_encoded wrapper for label encoding #3144
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
[ENH] Add load_classification_encoded wrapper for label encoding
Reference Issue
What does this implement/fix? Explain your changes.
This PR implements a new wrapper function,
load_classification_encoded, inaeon/datasets/_data_loaders.py.As discussed in issue #2807, this function allows users to load classification datasets with target labels automatically encoded as integers (using
sklearn.LabelEncoder) without modifying the signature of the baseload_classificationfunction.Key changes:
load_classification_encoded. It loads train/test splits, fits a label encoder on the training set, transforms both sets, and optionally returns the fitted encoder.aeon/datasets/__init__.py.aeon/datasets/tests/test_data_loaders.pyto verify integer conversion and encoder return behavior.Does your contribution introduce a new dependency? If yes, which one?
No
Any other comments?
I chose to implement this as a separate wrapper function rather than modifying
load_classificationarguments to avoid complicating the existing API signature, consistent with the maintainers' suggestions in the issue thread.PR checklist
For all contributions
For new estimators and functions
__maintainer__at the top of relevant files and want to be contacted regarding its maintenance. Unmaintained files may be removed. This is for the full file, and you should not add yourself if you are just making minor changes or do not want to help maintain its contents.For developers with write access