Module sktmls.models.contrib.sample_model

Classes

class SampleModel (model, model_name: str, model_version: str, features: List[str] = None)

단일 LightGBM을 사용하는 샘플 모델입니다.

모델을 추론하여 y score가 0.2 이상인 경우 결과를 반환합니다.

Ancestors

• MLSLightGBMModel
• MLSModel

Methods

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

Inherited members

• MLSLightGBMModel:
  • save