feat:VOV模型

XGB/vov_xgboost_train.py

@@ -27,24 +27,7 @@ column_names = ['曝光占比', 'vov0', '分子', '分母', '1_vov0', '2_vov0',
 
 # 创建一个logger
 logger = logging.getLogger("vov_xgboost_train.py")
-logger.setLevel(logging.INFO)  # 设置日志级别
-
-# 创建Handler用于输出到文件
-file_handler = logging.FileHandler('xgboost_train.log')
-file_handler.setLevel(logging.INFO)  # 设置日志级别为INFO
-
-# 创建Handler用于输出到控制台
-console_handler = logging.StreamHandler()
-console_handler.setLevel(logging.INFO)  # 设置日志级别为INFO
-
-# 定义日志格式
-formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-file_handler.setFormatter(formatter)
-console_handler.setFormatter(formatter)
-
-# 将Handler添加到Logger
-logger.addHandler(file_handler)
-logger.addHandler(console_handler)
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 
 
 def get_partition_df(table, dt):

config/ConfigManager.py

@@ -1,15 +1,19 @@
 import configparser
+import logging
 import os
 from pathlib import Path
 
+logger = logging.getLogger("ConfigManager.py")
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
 
 class ConfigManager(object):
     def __init__(self):
         # 获取环境变量 ROV_OFFLINE_ENV
         env = os.environ.get('ROV_OFFLINE_ENV')
-        print(f"当前环境: {env}")
+        logger.info(f"当前环境: {env}")
         project_home = Path(__file__).parent.parent.absolute()
-        print(f"当前项目目录为: {project_home}")
+        logger.info(f"当前项目目录为: {project_home}")
         self.env = env
         self.project_home = project_home
 

helper/RedisHelper.py

@@ -1,5 +1,10 @@
+import logging
+
 import redis
 
+logger = logging.getLogger("RedisHelper.py")
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
 
 class RedisHelper(object):
     def __init__(self, host: str, port: int, password: str):
@@ -9,8 +14,9 @@ class RedisHelper(object):
         self.redis_conn = redis.Redis(host=self.host, port=self.port, password=password)
 
     def add_number_to_set(self, key: str, number):
-        print(f"Redis Set写入: {key} ---> {number}")
+        logger.info(f"Redis Set写入: {key} ---> {number}")
         self.redis_conn.sadd(key, number)
 
     def set_expire(self, key: str, expire: int):
+        logger.info(f"Redis Expire: {key} ---> {expire}")
         self.redis_conn.expire(key, expire)