Skip to content

multi_label classification #8

@Amiri-Yossri

Description

@Amiri-Yossri

i want to classify keywords that i extracted from the text using Setfit, but I want multi_label classification because there are words that can define multiple classes and there are words that don't define any class. how i can do that. this is the code after defining my training words (I got an error in this second line):
clf = SetFitClassifier("paraphrase-MiniLM-L3-v2")
clf.fit(docs, labels, multi_label = True)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions