seanpedrickcase commited on
Commit
f42e3d1
·
1 Parent(s): 8216d8c

Re-added TruncatedSVD dependency to topic_core_funcs.py

Browse files
Files changed (1) hide show
  1. funcs/topic_core_funcs.py +2 -1
funcs/topic_core_funcs.py CHANGED
@@ -17,7 +17,8 @@ from funcs.helper_functions import read_file, zip_folder, delete_files_in_folder
17
  from funcs.embeddings import make_or_load_embeddings, torch_device
18
  from funcs.bertopic_vis_documents import visualize_documents_custom, visualize_hierarchical_documents_custom, hierarchical_topics_custom, visualize_hierarchy_custom
19
  from funcs.representation_model import create_representation_model, llm_config, chosen_start_tag, random_seed, RUNNING_ON_AWS
20
- from sklearn.feature_extraction.text import CountVectorizer
 
21
  import funcs.anonymiser as anon
22
  from umap import UMAP
23
 
 
17
  from funcs.embeddings import make_or_load_embeddings, torch_device
18
  from funcs.bertopic_vis_documents import visualize_documents_custom, visualize_hierarchical_documents_custom, hierarchical_topics_custom, visualize_hierarchy_custom
19
  from funcs.representation_model import create_representation_model, llm_config, chosen_start_tag, random_seed, RUNNING_ON_AWS
20
+ from sklearn.feature_extraction.text import CountVectorizer, TfidfVectorizer
21
+ from sklearn.decomposition import TruncatedSVD
22
  import funcs.anonymiser as anon
23
  from umap import UMAP
24