Skip to content

Returning all class probabilities #14

@nishkalavallabhi

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

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions