Views
No views yet
pip install sentence_transformers umap-learn bertopic1
2Then, load the model and encode your documents:
3
4```python
5from sentence_transformers import SentenceTransformer
6from umap import UMAP
7from bertopic import BERTopic
8
9# Load the sentence transformer model
10sentence_model = SentenceTransformer("all-MiniLM-L6-v2")
11
12# Set the random state in the UMAP model to prevent stochastic behavior
13umap_model = UMAP(n_neighbors=15, n_components=5, min_dist=0.0, metric='cosine', random_state=42)
14
15# Load the BERTopic model
16my_model = BERTopic.load("from/path/model.bin")
17
18# Encode your documents
19document_embeddings = sentence_model.encode(documents)1
2sentences = "my sentence"
3
4embeddings = sentence_model.encode([sentences])
5
6topic , _ =my_model.transform([sentences],embeddings)
7