How to output the predicted probability distribution of the classifier for various labels

facebookresearch / ParlAI

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.

https://parl.ai

MIT License

10.49k stars 2.1k forks source link

How to output the predicted probability distribution of the classifier for various labels #5009

Closed Oreki-PJ closed 1 year ago

Oreki-PJ commented 1 year ago

i run "parlai interactive --model-file zoo:style_gen/prev_curr_classifier/model --model projects.style_gen.classifier:ClassifierAgent --classes-from-file image_chat_personalities_file", and i can only get the result of the classifier. but how can get the output the predicted probability distribution of the classifier for various labels ?

Oreki-PJ commented 1 year ago

klshuster commented 1 year ago

if you set --print-scores True it will print the probability

github-actions[bot] commented 1 year ago

This issue has not had activity in 30 days. Please feel free to reopen if you have more issues. You may apply the "never-stale" tag to prevent this from happening.