Skip to content

Request for Text classification with Unsloth notebook for 10+ classes #192

@palette-knife25

Description

@palette-knife25

Hello! I want to apply unsloth_classification notebook for my task. However, I have about 400 classes, which are not represented by single tokens. I believe that the current version of the notebook is unfit for my task. Specifically, the parts with classification head trimming and Data Collator are raising my suspicion, since multi-digit classes are not represented by single tokens. Could you please add another version of unsloth_classification notebook that explaines how to work with classification for 10+ classes? Thank you!

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