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

Default behaviour in TextClassificationPipeline for regression problem type #33973

Open
subhalingamd opened this issue Oct 5, 2024 · 0 comments
Labels
Feature request Request for a new feature

Comments

@subhalingamd
Copy link

subhalingamd commented Oct 5, 2024

Feature request

The AutoModelForSequenceClassification class also supports problem_type=regression. I am not sure if it is popularly used but I believe that function_to_apply in TextClassificationPipeline should behave as if it is set to "none" for this case?

Motivation

Currently, the "sigmoid" function is also applied for regression tasks which, in my opinion, might not be the correct default behaviour.

Your contribution

I believe this can be handled by adding an additional condition for self.model.config.problem_type == "regression" before this line.

@subhalingamd subhalingamd added the Feature request Request for a new feature label Oct 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Feature request Request for a new feature
Projects
None yet
Development

No branches or pull requests

1 participant