Update BERTopic/my_topic_modeling.py

BERTopic/my_topic_modeling.py

@@ -414,6 +414,7 @@ def coherence_score(topic_model, df_documents):
 
 def working(args: argparse.Namespace, name_dataset: str):
     
+    
     ############# Create logger##################################
     fandc_logger = create_logger_file_and_console()
     file_logger = create_logger_file()
@@ -421,6 +422,7 @@ def working(args: argparse.Namespace, name_dataset: str):
     ##############################################################
     
     ######### Create dataframe for dataset booking and tripadvisor #####
+    fandc_logger.log(logging.INFO, f'STARTING WITH TOPIC MODEL FOR {name_dataset} dataset')
     fandc_logger.log(logging.INFO, f'Get data from {name_dataset}')
     doc_source = f'en_{name_dataset}.csv'
     list_tmp = prepare_data(doc_source, doc_type, type_framework = 'pandas')
@@ -585,7 +587,7 @@ def working(args: argparse.Namespace, name_dataset: str):
 
         fandc_logger.log(logging.INFO, f'Save topics reduce and topic over time for {name_dataset} dataset successfully!')
     ###################################
-    
+        fandc_logger.log(logging.INFO, f'ENDING TRAINING TOPIC MODELING {name_dataset} dataset\n')