easy-ml/model/model_base.py

# -*- coding:utf-8 -*-
"""
@author: yq
@time: 2024/1/2
@desc: 模型基类
"""
import abc

import pandas as pd

from entitys import DataFeatureEntity


class ModelBase(metaclass=abc.ABCMeta):

    @abc.abstractmethod
    def train(self, data: DataFeatureEntity, *args, **kwargs):
        pass

    @abc.abstractmethod
    def predict_prob(self, x: pd.DataFrame, *args, **kwargs):
        pass

    @abc.abstractmethod
    def predict(self, x: pd.DataFrame, *args, **kwargs):
        pass

    @abc.abstractmethod
    def export_model_file(self, ):
        pass