WiFddlZddlmZddlmZddlmZGddeZy)N)List) BaseEmbedder)select_backendceZdZdZfdZd deededejfdZ d deededejfdZ xZ S) WordDocEmbedderz,Combine a document- and word-level embedder.cbt|t||_t||_y)N)super__init__rembedding_modelword_embedding_model)selfr r __class__s f/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/bertopic/backend/_word_doc.pyr zWordDocEmbedder.__init__ s) -o>$23G$H!wordsverbosereturnc:|jj||S)aZEmbed a list of n words into an n-dimensional matrix of embeddings. Arguments: words: A list of words to be embedded verbose: Controls the verbosity of the process Returns: Word embeddings with shape (n, m) with `n` words that each have an embeddings size of `m` )r embed)r rrs r embed_wordszWordDocEmbedder.embed_wordss((..ug>>rdocumentc:|jj||S)ahEmbed a list of n words into an n-dimensional matrix of embeddings. Arguments: document: A list of documents to be embedded verbose: Controls the verbosity of the process Returns: Document embeddings with shape (n, m) with `n` documents that each have an embeddings size of `m` )r r)r rrs rembed_documentszWordDocEmbedder.embed_documentss##))(G<r(s/2$=l$=r