Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

BUG: Fix feature transformation #79

Merged
merged 7 commits into from
Nov 1, 2024
Merged

Conversation

AnirudhDagar
Copy link
Collaborator

@AnirudhDagar AnirudhDagar commented Oct 31, 2024

Issue:

KeyError: '[None] not found in axis'
/home/ubuntu/new_oct23/autogluon-assistant/src/autogluon_assistant/transformer/feature_transformers/base.py(54)transform()
-> test_x = task.test_data.drop(columns=[task.test_id_column])

Description of changes:

When the datasets (specifically DS-Agent 17 datasets) contain label column in test data, the column was not being dropped before transformation. Also the test_id_column was being used for train set. That has been fixed.

NOTE: This PR needs careful review, please only approve after that (I'll merge it after a few more tests). Currently it was tested with toy_data inside the repo.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@AnirudhDagar AnirudhDagar marked this pull request as ready for review November 1, 2024 00:44
Copy link
Contributor

@FANGAreNotGnu FANGAreNotGnu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@AnirudhDagar AnirudhDagar merged commit c06d55f into main Nov 1, 2024
1 check passed
@AnirudhDagar AnirudhDagar deleted the fix_feature_transform branch November 1, 2024 03:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants