Use logger

backend/training.py

@@ -18,6 +18,8 @@ from settings.config import settings
 
 
 logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 config_info = settings.CONFIG_INFO
 
 

backend/validation.py

@@ -18,6 +18,8 @@ from settings.config import settings
 
 
 logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
 config_info = settings.CONFIG_INFO
 
 
@@ -172,4 +174,4 @@ if __name__ == '__main__':
                                              step_length=config_info['buffer_length'])
     fig_erds.savefig(os.path.join(data_dir, 'erds.pdf'))
     fig_pred.savefig(os.path.join(data_dir, 'pred.pdf'))   
-    logging.info(f'precision: {metrics[0]:.4f}, recall: {metrics[1]:.4f}, f_beta_score: {metrics[2]:.4f}, corr: {metrics[3]:.4f}')
+    logger.info(f'precision: {metrics[0]:.4f}, recall: {metrics[1]:.4f}, f_beta_score: {metrics[2]:.4f}, corr: {metrics[3]:.4f}')