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")}
