lightgbm_score/applications/functions.py

"""
@author: luojunhui
"""
import asyncio

from .model_init import models


class ParamProcess(object):
    """
    处理 params， 继承 models
    """
    def __init__(self):
        self.model_v1 = models.model_v1
        self.model_v2 = models.model_v2
        self.layer_encoder = models.label_encoder

    async def predict_score(self, version, features):
        """
        预测
        :param version: 模型版本
        :param features: 视频被 label_encoder 之后的features
        :return: score: 返回的分数
        """
        match version:
            case "v1":
                return await self.model_v1.predict(features)
            case "v2":
                return await self.model_v2.predict(features)

    async def process_label(self, params):
        """
        处理类别 features 和 float features
        :param params: 接收到的参数
        :return:
        """
        version = params['version']
        features = params['features']
        match version:
            case "v1":
                # 全部转化为类别
                print("all to string cate")
                # features = []
                return version, features
            case "v2":
                print("all to float cate")
                # features = []
                return version, features

    async def process(self, params):
        """
        处理
        :param params:
        :return:
        """
        version, features = await self.process_label(params)
        print(version, features)
        # return await self.predict_score(version, features)