Skip to content
This repository was archived by the owner on Nov 3, 2023. It is now read-only.

Allow the style classifier to save output probabilities #4026

Merged
merged 1 commit into from
Sep 17, 2021

Conversation

EricMichaelSmith
Copy link
Contributor

@EricMichaelSmith EricMichaelSmith commented Sep 17, 2021

Patch description
If --print-scores True, allow the classifier from the style-controlled generation project to save per-class probabilities in a {class: prob} dict. Useful for looking at the per-example per-class probabilities when doing parlai eval_model

Testing steps
Existing CI checks

Copy link
Contributor

@klshuster klshuster left a comment

Choose a reason for hiding this comment

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

cool, this seems quite useful

@EricMichaelSmith EricMichaelSmith merged commit 3a97edf into main Sep 17, 2021
@EricMichaelSmith EricMichaelSmith deleted the save-probs branch September 17, 2021 21:50
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants