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

change threshold value in predict for multi-label classification? #1562

Open
lececefifi opened this issue Jan 23, 2024 · 1 comment
Open

Comments

@lececefifi
Copy link

Hi - I am using the multilabel model classification, I have fine-tuned my models with different threshold values, and I would like to also be able to change the threshold value when I run predict function. I tried adding args to the function but no change is seen. Is this the right way or is there a different way to do this? Or this is not currently supported?

Thank you,
Cecilia

Is your feature request related to a problem? Please describe.
A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]

Describe the solution you'd like
A clear and concise description of what you want to happen.

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.

Additional context
Add any other context or screenshots about the feature request here.

@lececefifi
Copy link
Author

I have the same issue as this previous post, #252

I have tried setting it in the model.predict(),
model_args = {
"threshold": 0.70
}

I have also added the same when model is loaded,
model = MultiLabelClassificationModel(
model_type,
model_name,
args = model_args,
use_cuda=cuda_available,

can someone point me to the right direction as where I have missed? Thank you!

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

No branches or pull requests

1 participant