Update BERTopic/my_topic_modeling.py

BERTopic/my_topic_modeling.py

@@ -171,6 +171,15 @@ def init_args():
         required=True,
         help="Only train both models for calculating coherence score.",
     )
+
+    parser.add_argument(
+        "--need_reduce_n_topics",
+        default="yes",
+        type=str,
+        required=True,
+        help="Need reduce n topics and show topic modeling over timestamp with this.",
+    )
+
     
     args = parser.parse_args()
 
@@ -559,7 +568,7 @@ def working(args: argparse.Namespace, name_dataset: str):
 
         # ****** 2
         # Get reduce topics and topic over time for each n_topics
-    if args.need_reduce_n_tops == 'yes':
+    if args.need_reduce_n_topics == 'yes':
         fandc_logger.log(logging.INFO, f'Staring reduce topics and topic over time from 10 to 50...')
         for n_topics in [10,20,30,40,50]:
             topic_model_copy = copy.deepcopy(topic_model)