Tyler Burns commited on
Commit
3952d5a
1 Parent(s): 341cb78

switched back to title for nltk keywords

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. flycheck_app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ print("running sentence embeddings...")
30
  # model_name = 'all-mpnet-base-v2'
31
  model_name = 'all-MiniLM-L6-v2'
32
  model = SentenceTransformer(model_name)
33
- sentence_embeddings = model.encode(md['body'].tolist(), show_progress_bar = True)
34
  sentence_embeddings = pd.DataFrame(sentence_embeddings)
35
 
36
  # Reduce dimensionality
 
30
  # model_name = 'all-mpnet-base-v2'
31
  model_name = 'all-MiniLM-L6-v2'
32
  model = SentenceTransformer(model_name)
33
+ sentence_embeddings = model.encode(md['title'].tolist(), show_progress_bar = True)
34
  sentence_embeddings = pd.DataFrame(sentence_embeddings)
35
 
36
  # Reduce dimensionality
flycheck_app.py CHANGED
@@ -30,7 +30,7 @@ print("running sentence embeddings...")
30
  # model_name = 'all-mpnet-base-v2'
31
  model_name = 'all-MiniLM-L6-v2'
32
  model = SentenceTransformer(model_name)
33
- sentence_embeddings = model.encode(md['body'].tolist(), show_progress_bar = True)
34
  sentence_embeddings = pd.DataFrame(sentence_embeddings)
35
 
36
  # Reduce dimensionality
 
30
  # model_name = 'all-mpnet-base-v2'
31
  model_name = 'all-MiniLM-L6-v2'
32
  model = SentenceTransformer(model_name)
33
+ sentence_embeddings = model.encode(md['title'].tolist(), show_progress_bar = True)
34
  sentence_embeddings = pd.DataFrame(sentence_embeddings)
35
 
36
  # Reduce dimensionality