You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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!
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.
The text was updated successfully, but these errors were encountered: