Module sktmls.models.contrib.tw_random_greeting_rule_model

Classes

class TwRandomGreetingRuleModel (model_name: str, model_version: str, image_type_ids: List[str], text_type_ids: List[str], greeting_ids: List[str], greeting_ids_shuffle: bool = False)

MLS 모델 레지스트리에 등록되는 Rule 기반 클래스입니다.

TW 그리팅 추천 모델에 특화된 모델이며, 2개의 그리팅 타입(이미지 / 텍스트) 및 1개의 그리팅 랭킹을 사용합니다.

Args

• model_name: (str) 모델 이름
• model_version: (str) 모델 버전
• image_type_ids: (list) 티월드그리팅 이미지 타입
• text_type_ids: (list) 티월드그리팅 텍스트 타입
• greeting_ids: (list) 티월드그리팅 랭킹
• greeting_ids_shuffle: (bool) 티월드그리팅 랭킹 Shuffle 여부

Example

random_greeting_rule_model_v1 = TwRandomGreetingRuleModel(
    model_name="random_greeting_rule_model",
    model_version="v1",
    image_type_ids=["A", "B"],
    text_type_ids=["A", "B", "C"],
    greeting_ids=["C1", "C2", "C3", "C4", "C5"],
    greeting_ids_shuffle=True
)

result = random_greeting_rule_model_v1.predict(None)

Ancestors

• MLSRuleModel
• MLSModel

Methods

def predict(self, x: List[Any], **kwargs) ‑> Dict[str, Any]

Inherited members

• MLSRuleModel:
  • save