modify: 优化包引用路径

data_loader/__init__.py → loader/data/__init__.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 """
 @author: yq
-@time: 2024/10/30
+@time: 2024/11/1
 @desc: 
 """
 

data_loader/data_loader_mysql.py → loader/data/data_loader_mysql.py

@@ -8,7 +8,7 @@ import pandas as pd
 import pymysql
 
 from commom.logger import get_logger
-from data_loader.data_loader_base import DataLoaderBase
+from .data_loader_base import DataLoaderBase
 
 logger = get_logger()
 
@@ -23,7 +23,7 @@ class DataLoaderMysql(DataLoaderBase):
         self.conn = None
 
     def get_connect(self):
-        # TODO 后续改成线程池
+        #  TODO 后续改成线程池
         if self.conn == None:
             self.conn = pymysql.connect(host=self.host, port=self.port, user=self.user, passwd=self.passwd,
                                         db=self.db)

metric/__init__.py

@@ -4,6 +4,10 @@
 @time: 2022/10/24
 @desc: 指标计算相关
 """
+from .metric_base import MetricBase
+from .metric_by_sql_general import MetricBySqlGeneral
+
+__all__ = ['MetricBase', 'MetricBySqlGeneral']
 
 if __name__ == "__main__":
     pass

metric/metric_by_sql_general.py

@@ -6,7 +6,7 @@
 """
 import pandas as pd
 
-from data_loader.data_loader_base import DataLoaderBase
+from loader import DataLoaderBase
 from metric.metric_base import MetricBase
 
 
@@ -27,7 +27,7 @@ class MetricBySqlGeneral(MetricBase):
 
 
 if __name__ == "__main__":
-    from data_loader.data_loader_mysql import DataLoaderMysql
+    from loader import DataLoaderMysql
 
     data_loader = DataLoaderMysql(host="101.126.81.2", port=18001, user="root", passwd="Cqrcb2024", db="test")
     metric_clzz = MetricBySqlGeneral(metric_name="auc", metric_code="auc")

trainer/__init__.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+"""
+@author: yq
+@time: 2024/11/1
+@desc: 
+"""
+
+if __name__ == "__main__":
+    pass