modify: 优化包引用路径

loader/__init__.py → data/__init__.py

@@ -4,8 +4,8 @@
 @time: 2024/10/30
 @desc: 
 """
-from .data.data_loader_base import DataLoaderBase
-from .data.data_loader_mysql import DataLoaderMysql
+from .loader.data_loader_base import DataLoaderBase
+from .loader.data_loader_mysql import DataLoaderMysql
 
 __all__ = ['DataLoaderBase', 'DataLoaderMysql']
 

metric/metric_by_sql_general.py

@@ -6,7 +6,7 @@
 """
 import pandas as pd
 
-from loader import DataLoaderBase
+from data import DataLoaderBase
 from .metric_base import MetricBase
 
 

model/__init__.py

@@ -4,3 +4,10 @@
 @time: 2023/12/28
 @desc: 模型相关
 """
+
+from .model_base import ModelBase
+
+__all__ = ['ModelBase',]
+
+if __name__ == "__main__":
+    pass

test.py

@@ -4,15 +4,11 @@
 @time: 2024/11/1
 @desc: 
 """
-
+from data import DataLoaderMysql
 from metric import MetricBySqlGeneral
 
-
 if __name__ == "__main__":
-    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")
     metric = metric_clzz.calculate(data_loader, "select * from test.t1")
     print(metric.head(5))
-