Wi jddlZddlmZddlmZmZddlmZddlm Z m Z m Z ddl m Z Gdde Zy) N)tqdm)UnionList)Sentence)DocumentEmbeddingsTokenEmbeddingsDocumentPoolEmbeddings) BaseEmbeddercbeZdZdZdeeefffd Zddee de de jfdZ xZS) FlairBackenda[Flair Embedding Model. The Flair embedding model used for generating document and word embeddings. Arguments: embedding_model: A Flair embedding model Examples: ```python from bertopic.backend import FlairBackend from flair.embeddings import WordEmbeddings, DocumentPoolEmbeddings # Create a Flair Embedding model glove_embedding = WordEmbeddings('crawl') document_glove_embeddings = DocumentPoolEmbeddings([glove_embedding]) # Pass the Flair model to create a new backend flair_embedder = FlairBackend(document_glove_embeddings) ``` embedding_modelct|t|trt |g|_yt|t rd|jvrd|_||_ytd)N fine_tuneFzPlease select a correct Flair model by either using preparing a token or document embedding model: `from flair.embeddings import TransformerDocumentEmbeddings` `roberta = TransformerDocumentEmbeddings('roberta-base')`) super__init__ isinstancerr r r__dict__r ValueError)selfr __class__s c/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/bertopic/backend/_flair.pyrzFlairBackend.__init__!sj  o 7#9?:K#LD ); <o666,1)#2D L  documentsverbosereturncg}t|| D]}} |r t|n td}|jj||j j jj}|j|tj|}|S#t$r)td}|jj|YwxYw)aEmbed a list of n documents/words into an n-dimensional matrix of embeddings. Arguments: documents: A list of documents or words to be embedded verbose: Controls the verbosity of the process Returns: Document/words embeddings with shape (n, m) with `n` documents/words that each have an embeddings size of `m` )disablezan empty document) rrr embed RuntimeError embeddingdetachcpunumpyappendnpasarray)rrr embeddingsdocumentsentencer s rrzFlairBackend.embed7s YG < )H 5198H-xH[?\$$**84!**113779??AI   i ( )ZZ +    5#$78$$**84 5s3B((/CC)F)__name__ __module__ __qualname____doc__rrrrrstrboolr%ndarrayr __classcell__)rs@rr r sA,o?Q.Q(R,tCy4BJJrr )r#r%rtypingrr flair.datarflair.embeddingsrrr bertopic.backendr r rrr7s(XX)D<Dr