def get_feature_config(cls) -> Dict[str, rlt.IdFeatureConfig]: return { "a_id": rlt.IdFeatureConfig(feature_id=2002, id_mapping_name="a_mapping"), "b_id": rlt.IdFeatureConfig(feature_id=2003, id_mapping_name="b_mapping"), }
def feature_config(self): return rlt.ModelFeatureConfig( id_mapping_config={ "page": rlt.IdMapping(ids=list(range(100, 100 + self.embedding_size))) }, id_list_feature_configs=[ rlt.IdFeatureConfig( name="page_id", feature_id=2002, id_mapping_name="page" ) ], )
def get_feature_config(cls) -> Dict[str, rlt.IdFeatureConfig]: return {"page_id": rlt.IdFeatureConfig(feature_id=2002, id_mapping_name="page")}