Skip to content

Returning all class probabilities #14

Description

@nishkalavallabhi

Is there a way to return all class probabilities instead of just the most probable class? I tried changing the classifier call by adding return_all_scores=True (e.g., results = classifier("I love this package!", return_all_scores=True)) or using top_k=5 instead (to see top 5 probabilties) in the pipeline object. In both cases, I see an error "ValueError: can only convert an array of size 1 to a Python scalar" .

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions