Update streamlit_app_LDA.py

streamlit_app_LDA.py

@@ -23,6 +23,21 @@ num_topics = st.slider("Number of Topics", min_value=2, max_value=20, value=5)
 num_passes = st.slider("Number of Passes", min_value=5, max_value=50, value=10)
 seed_value = st.number_input("Random Seed", value=42)
 
+st.subheader("Beta -- dispersion of words in a topic - lower means less words in each topic")
+is_set_beta = st.checkbox("Set custom Beta (default: 1 / num_topics)? ")
+if is_set_beta:
+    beta = st.number_input("Beta", min_value=0.0, max_value=1.0, value=1/num_topics, step=0.05, format="%.3f")
+else:
+    beta = 1 / num_topics
+
+st.subheader("Alpha -- dispersion of topics in a document - lower means less topics in each document")
+is_set_alpha = st.checkbox("Set custom Alpha (default: dynamic per document)? ")
+if is_set_alpha:
+    alpha = st.number_input("Alpha", min_value=0.0, max_value=1.0, value=1/num_topics, step=0.05, format="%.3f")
+else:
+    alpha = "auto"
+
+
 st.header("Display")
 num_top_elements_to_show = st.slider("# top element to show in a topic", min_value=2, max_value=15, value=5)
 show_long_elements = st.checkbox("Show full element name")
@@ -93,6 +108,8 @@ lda_model = models.LdaModel(corpus,
                             num_topics=num_topics, 
                             id2word=dictionary, 
                             passes=num_passes, 
+                            eta=beta,
+                            alpha=alpha,
                             random_state=seed_value)
 
 # ---------------------------