easy-ml/entitys/__init__.py

# -*- coding: utf-8 -*-
"""
@author: yq
@time: 2024/10/30
@desc: 数据实体类
"""
from .train_config_entity import TrainConfigEntity
from .data_process_config_entity import DataProcessConfigEntity
from .data_feaure_entity import DataFeatureEntity, DataSplitEntity, DataPreparedEntity
from .db_config_entity import DbConfigEntity
from .metric_config_entity import MetricConfigEntity
from .metric_entity import MetricTrainEntity, MetricFucEntity
from .monitor_metric_config_entity import MonitorMetricConfigEntity

__all__ = ['DataFeatureEntity', 'DbConfigEntity', 'MetricTrainEntity', 'MonitorMetricConfigEntity', 'MetricConfigEntity',
           'MetricFucEntity', 'DataSplitEntity', 'DataProcessConfigEntity', 'TrainConfigEntity', 'DataPreparedEntity']

if __name__ == "__main__":
    pass