HarryLee commited on
Commit
4901a26
·
1 Parent(s): 9118322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ st.markdown("<h1 style='text-align: center;'>Topic Trendy</h1>", unsafe_allow_ht
16
  sentence_model = SentenceTransformer(model_name)
17
  umap_model = UMAP(n_neighbors=15, n_components=2, min_dist=0.1, metric="cosine")
18
  hdbscan_model = HDBSCAN(min_cluster_size=5, min_samples = 3, metric="euclidean", prediction_data=True)
19
- vectorizer_model = CountVectorizer(lowercase = True, ngram_range=(1, 3), analyzer="word", max_df=1.0, min_df=, stop_words="english")
20
 
21
  kw_model = BERTopic(embedding_model=sentence_model, umap_model = umap_model, hdbscan_model = hdbscan_model, vectorizer_model = vectorizer_model, nr_topics = "auto", calculate_probabilities = True)
22
  BerTopic_model = kw_model
 
16
  sentence_model = SentenceTransformer(model_name)
17
  umap_model = UMAP(n_neighbors=15, n_components=2, min_dist=0.1, metric="cosine")
18
  hdbscan_model = HDBSCAN(min_cluster_size=5, min_samples = 3, metric="euclidean", prediction_data=True)
19
+ vectorizer_model = CountVectorizer(lowercase = True, ngram_range=(1, 3), analyzer="word", max_df=1.0, min_df=0.5 stop_words="english")
20
 
21
  kw_model = BERTopic(embedding_model=sentence_model, umap_model = umap_model, hdbscan_model = hdbscan_model, vectorizer_model = vectorizer_model, nr_topics = "auto", calculate_probabilities = True)
22
  BerTopic_model = kw_model