edugp commited on
Commit
ab846df
1 Parent(s): ee732fe

Format file

Browse files
Files changed (1) hide show
  1. perplexity_lenses/engine.py +6 -1
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 = ["distiluse-base-multilingual-cased-v1", "distiluse-base-multilingual-cased-v2", "all-mpnet-base-v2", "flax-sentence-embeddings/all_datasets_v3_mpnet-base"]
 
 
 
 
 
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