haystack TopicClassifier

ylamdve6 于 4个月前发布在其他

关注(0)|答案(1)|浏览(41)

将输入的文本/文档归类到一个列表中的任意类别。它与LanguageClassifier非常相似，但更通用，因为它可以根据任意类别进行分类。可以使用零样本文本分类器实现。同样，我们可能需要文本和文档版本。
TextTopicClassifier的草稿I/O:

@component
class TextTopicClassifier:
    
    def __init__(self, topics: Optional[List[str]] = None):
        self.topics = topics
        component.set_output_types(**{topic: List[str] for topic in topics})

    def run(self, strings: List[str]):
        # classify the strings
        return {"topic_1": strings_1, "topic_2": strings_2, ...}

DocumentTopicClassifier的草稿I/O:

@component
class DocumentTopicClassifier:
    
    def __init__(self, topics: Optional[List[str]] = None):
        self.topics = topics
        component.set_output_types(**{topic: List[str] for topic in topics})

    def run(self, documents: List[Document]):
        # classify the documents
        return {"topic_1": docs_1, "topic_2": docs_2, ...}

您正在编辑的内容已更改。请复制您的编辑并刷新页面。