Spaces:
Runtime error
Runtime error
Format file
Browse files
perplexity_lenses/engine.py
CHANGED
|
@@ -12,7 +12,12 @@ from sentence_transformers import SentenceTransformer
|
|
| 12 |
from perplexity_lenses.visualization import draw_interactive_scatter_plot
|
| 13 |
|
| 14 |
logger = logging.getLogger(__name__)
|
| 15 |
-
EMBEDDING_MODELS = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
DIMENSIONALITY_REDUCTION_ALGORITHMS = ["UMAP", "t-SNE"]
|
| 17 |
DOCUMENT_TYPES = ["Whole document", "Sentence"]
|
| 18 |
SEED = 0
|
|
|
|
| 12 |
from perplexity_lenses.visualization import draw_interactive_scatter_plot
|
| 13 |
|
| 14 |
logger = logging.getLogger(__name__)
|
| 15 |
+
EMBEDDING_MODELS = [
|
| 16 |
+
"distiluse-base-multilingual-cased-v1",
|
| 17 |
+
"distiluse-base-multilingual-cased-v2",
|
| 18 |
+
"all-mpnet-base-v2",
|
| 19 |
+
"flax-sentence-embeddings/all_datasets_v3_mpnet-base",
|
| 20 |
+
]
|
| 21 |
DIMENSIONALITY_REDUCTION_ALGORITHMS = ["UMAP", "t-SNE"]
|
| 22 |
DOCUMENT_TYPES = ["Whole document", "Sentence"]
|
| 23 |
SEED = 0
|