WikjddlZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZGddeZy) N)pipeline)Pipeline) csr_matrix)MappingListTupleAny)BaseRepresentationc eZdZdZdidfdeededeeefdefdZ d e jd e d eeee eeffd eeee eefffd Zy)ZeroShotClassificationaZero-shot Classification on topic keywords with candidate labels. Arguments: candidate_topics: A list of labels to assign to the topics if they exceed `min_prob` model: A transformers pipeline that should be initialized as "zero-shot-classification". For example, `pipeline("zero-shot-classification", model="facebook/bart-large-mnli")` pipeline_kwargs: Kwargs that you can pass to the transformers.pipeline when it is called. NOTE: Use `{"multi_label": True}` to extract multiple labels for each topic. min_prob: The minimum probability to assign a candidate label to a topic Usage: ```python from bertopic.representation import ZeroShotClassification from bertopic import BERTopic # Create your representation model candidate_topics = ["space and nasa", "bicycles", "sports"] representation_model = ZeroShotClassification(candidate_topics, model="facebook/bart-large-mnli") # Use the representation model in BERTopic on top of the default pipeline topic_model = BERTopic(representation_model=representation_model) ``` zfacebook/bart-large-mnlig?candidate_topicsmodelpipeline_kwargsmin_probc||_t|trtd||_n#t|t r||_n t d||_||_y)Nzzero-shot-classification)rzwMake sure that the HF model that youpass is either a string referring to aHF model or a `transformers.pipeline` object.) r isinstancestrrrr ValueErrorrr)selfr rrrs m/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/bertopic/representation/_zeroshot.py__init__zZeroShotClassification.__init__&sZ!1 eS !!" &u ##'' 6$&!$'x(@.QYBZ$[ALE5t}},)00%@A  )!,t}}<&4X&>q&A>RZC[\]C^%_$`!$%*$*5M!&'",!uR#FWBX=X7Y%Z!g%ZZ!$5N5 !' 6*5^.&[s *E" E'N)__name__ __module__ __qualname____doc__rrrr floatrpd DataFramerrr6rrr r s>0-/ !s)!!!c* !  !*-<<- - T%U "3445 - d5e,-- . -rr )pandasr< transformersrtransformers.pipelines.baser scipy.sparsertypingrrrr bertopic.representation._baser r r>rrrEs)!0#,,<_/_r