easy-ml/test.py

# -*- coding: utf-8 -*-
"""
@author: yq
@time: 2024/11/1
@desc: 
"""

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))